Searched refs:kernel_side_table (Results 1 – 6 of 6) sorted by relevance
82 kernel_side_table = KernelSideTable() variable564 kernel = kernel_side_table.get_kernel(kernel_idx)565 constant_args = kernel_side_table.get_constant_args(constant_args_idx)632 kernel = kernel_side_table.get_kernel(kernel_idx)633 constant_args = kernel_side_table.get_constant_args(constant_args_idx)804 variable.kernel_idx = kernel_side_table.add_kernel(kernel)1004 constant_args_idx = kernel_side_table.add_constant_args(non_graphable_args)
69 from torch._higher_order_ops.triton_kernel_wrap import kernel_side_table71 add_kernel_id = kernel_side_table.add_kernel(add_kernel)81 constant_args_idx = kernel_side_table.add_constant_args(119 from torch._higher_order_ops.triton_kernel_wrap import kernel_side_table126 kernel_side_table.reset_table()130 kernel_idx=kernel_side_table.add_kernel(mul2_kernel),131 constant_args_idx=kernel_side_table.add_constant_args(171 from torch._higher_order_ops.triton_kernel_wrap import kernel_side_table202 kernel_idx=kernel_side_table.add_kernel(mul2_inplace_kernel),203 constant_args_idx=kernel_side_table.add_constant_args([all …]
824 from torch._higher_order_ops.triton_kernel_wrap import kernel_side_table831 self.kernel_idx = kernel_side_table.add_kernel(kernel)964 kernel_side_table,982 constant_args_idx = kernel_side_table.add_constant_args(constant_args)
11 kernel_side_table,633 kernel = kernel_side_table.get_kernel(kernel_idx)
4651 from torch._higher_order_ops.triton_kernel_wrap import kernel_side_table4653 kernel = kernel_side_table.get_kernel(self.kernel_idx)
6214 from torch._higher_order_ops.triton_kernel_wrap import kernel_side_table6216 constant_args = kernel_side_table.get_constant_args(constant_args_idx)