The WebGPU intrinsics submodule.
| 2980 | |
| 2981 | |
| 2982 | class WebGPUNamespace: |
| 2983 | """The WebGPU intrinsics submodule.""" |
| 2984 | |
| 2985 | @staticmethod |
| 2986 | def subgroup_shuffle(var, lane): |
| 2987 | if isinstance(var, Buffer): |
| 2988 | var = var[0] |
| 2989 | return _tir_op.call_intrin(var.dtype, "tirx.webgpu.subgroup_shuffle", var, lane) |
| 2990 | |
| 2991 | @staticmethod |
| 2992 | def subgroup_shuffle_up(var, delta): |
| 2993 | if isinstance(var, Buffer): |
| 2994 | var = var[0] |
| 2995 | return _tir_op.call_intrin(var.dtype, "tirx.webgpu.subgroup_shuffle_up", var, delta) |
| 2996 | |
| 2997 | @staticmethod |
| 2998 | def subgroup_shuffle_down(var, delta): |
| 2999 | if isinstance(var, Buffer): |
| 3000 | var = var[0] |
| 3001 | return _tir_op.call_intrin(var.dtype, "tirx.webgpu.subgroup_shuffle_down", var, delta) |
| 3002 | |
| 3003 | |
| 3004 | webgpu = WebGPUNamespace() |
no outgoing calls
no test coverage detected
searching dependent graphs…