Home
last modified time | relevance | path

Searched refs:ExternKernel (Results 1 – 12 of 12) sorted by relevance

/external/pytorch/torch/_inductor/kernel/
Dconv.py428 x = ir.ExternKernel.require_stride_order(x, channels_last_order(rank))
538 x = ir.ExternKernel.require_channels_last(x)
541 weight = ir.ExternKernel.require_channels_last(weight)
548 x = ir.ExternKernel.require_stride_order(x, req_stride_order)
549 weight = ir.ExternKernel.require_stride_order(weight, req_stride_order)
Dbmm.py135 t = ir.ExternKernel.require_contiguous(t)
Dflex_attention.py17 ExternKernel,
1709 delta = ExternKernel.require_contiguous(delta)
Dflex_decoding.py448 query = ir.ExternKernel.realize_input(query)
/external/pytorch/torch/_inductor/
Dgraph.py1057 result = [ir.ExternKernel.realize_input(x) for x in result]
1246 result = ir.ExternKernel.require_stride_order(result, stride_order)
1277 result = ir.ExternKernel.require_stride_order(
1327 result = ir.ExternKernel.require_stride_order(
1336 result = ir.ExternKernel.require_stride_order(
Dir.py2179 x = ExternKernel.realize_input(x)
3754 x = ExternKernel.realize_input(x)
3944 class ExternKernel(InputsKernel): class
4544 class ExternKernelOut(ExternKernel):
4607 class ExternKernelAlloc(ExternKernel):
4647 class UserDefinedTritonKernel(ExternKernel):
4776 class MutationOutput(ExternKernel):
4799 class InplaceBernoulliFallback(ExternKernel):
4844 class InplaceCopyFallback(ExternKernel):
4895 class MutatingFirstArgExternKernel(ExternKernel):
[all …]
Dmkldnn_lowerings.py180 b = ir.ExternKernel.realize_input(b)
243 b = ir.ExternKernel.realize_input(b)
Dlowering.py1956 return ir.ExternKernel.require_stride_order(input_tensor, stride_order)
2091 ir.IRNode, ir.ExternKernel.require_stride1, (args, kwargs)
2098 ir.IRNode, ir.ExternKernel.require_contiguous, (args, kwargs)
2105 ir.IRNode, ir.ExternKernel.require_channels_last, (args, kwargs)
2114 return ir.ExternKernel.require_stride_order(arg, stride_order)
2178 return ir.ExternKernel.require_stride_order(arg, stride_order)
2201 ir.ExternKernel.realize_input(arg), meta_stride
2213 ir.ExternKernel.realize_input(arg), meta_stride
2216 return ir.ExternKernel.require_stride_order(arg, stride_order)
2912 return ir.ExternKernel.require_stride_order(x, stride_order)
[all …]
Dselect_algorithm.py1738 return ir.ExternKernel.require_stride1(ir.ExternKernel.realize_input(args[0]))
Dscheduler.py647 assert isinstance(self.node, ir.ExternKernel), f"{type(self.node)=}"
1857 elif isinstance(node, ir.ExternKernel):
3329 assert isinstance(node, ir.ExternKernel), f"{type(node)=}"
Dutils.py496 elif isinstance(node_schedule, ir.ExternKernel):
/external/pytorch/test/distributed/
Dtest_compute_comm_reordering.py42 elif isinstance(snode.node, ir.ExternKernel):