| D | triton_kernel_wrap.py | 528 def __call__(self, kernel_idx, constant_args_idx, grid, kwargs): argument
545 def __call__(self, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone): argument
560 *, kernel_idx, constant_args_idx, grid, kwargs argument
582 mode, *, kernel_idx, constant_args_idx, grid, kwargs argument
589 def _(*, kernel_idx, constant_args_idx, grid, kwargs): argument
611 mode, *, kernel_idx, constant_args_idx, grid, kwargs argument
629 ctx, kernel_idx, constant_args_idx, grid, kwargs argument
667 *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
688 mode, *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
704 mode, *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
[all …]
|