Home
last modified time | relevance | path

Searched defs:IRNode (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A Dcuda_kernel.py60 def arg_name(self, node: IRNode) -> Optional[str]:
70 def check_not_null(self, node: IRNode) -> str:
101 inputs: List[IRNode],
102 outputs: List[IRNode],
190 def dtype(self, node: IRNode) -> Optional[str]:
199 def cutlass_dtype(self, node: IRNode, default_dtype="void") -> Optional[str]:
207 def max_valid_index(self, node: IRNode, default=-1):
216 def offset(self, node: IRNode) -> str:
225 def ptr(self, node: IRNode) -> str:
240 node: IRNode,
[all …]
H A Dgemm_template.py627 bias: IRNode,
1219 X: IRNode,
1220 W: IRNode,
1221 Bias: IRNode,
1222 Y: IRNode,
1269 def clone_with_transposed_stride(node: IRNode) -> IRNode:
1499 X: IRNode,
1500 W: IRNode,
1501 Bias: IRNode,
1502 Meta: IRNode,
[all …]
H A Dcuda_template.py109 epilogue_nodes: Optional[List[IRNode]] = None,
246 def cutlass_type_cast(self, node: IRNode, ptr: str) -> str:
252 def cutlass_sparse_meta_type_cast(self, node: IRNode, ptr: str) -> str:
H A Dcutlass_epilogue_gen.py70 epilogue_nodes: List[IRNode],
251 epilogue_nodes: List[IRNode],
/aosp_15_r20/prebuilts/go/linux-x86/src/cmd/compile/internal/pgoir/
Dirgraph.go73 type IRNode struct { struct
75 AST *ir.Func
78 LinkerSymbolName string
82 OutEdges map[pgo.NamedCallEdge]*IREdge
86 func (i *IRNode) Name() string {
208 func createIRGraphEdge(fn *ir.Func, callernode *IRNode, name string, namedEdgeMap pgo.NamedEdgeMap,…
237 func addIREdge(callerNode *IRNode, callerName string, call ir.Node, callee *ir.Func, namedEdgeMap p…
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcomm_analysis.py40 def get_collective_type(node: ir.IRNode) -> NCCL_COLL:
56 def get_collective_input_size_bytes(node: ir.IRNode) -> int:
69 def get_collective_group_size(node: ir.IRNode) -> int:
161 def estimate_nccl_collective_runtime(node: ir.IRNode) -> float:
H A Ddebug.py530 input_nodes: List[ir.IRNode],
539 def build_node_info(node: ir.IRNode) -> Dict[str, str]:
H A Dselect_algorithm.py548 def call_kernel(self, name: str, node: Optional[ir.IRNode] = None):
1551 input_nodes: List[ir.IRNode],
H A Dir.py274 class IRNode: class
681 input_node: Optional[IRNode] = None,
948 input_node: Optional[IRNode] = None,
2942 def __init__(self, target: IRNode):
3574 mutated_inputs: Optional[Iterable[IRNode]] = None,
3661 inputs: List[IRNode],
4761 def mark_node_as_mutating(cur_buffer, *mutated_nodes: IRNode):
7623 predicate: IRNode,
H A Dlowering.py5464 def get_constant_value(x: ir.IRNode) -> Optional[ir.Constant]:
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A Drocm_kernel.py52 def arg_name(self, node: IRNode) -> Optional[str]:
64 inputs: List[IRNode],
65 outputs: List[IRNode],
186 make_kernel_render: Callable[[ROCmTemplateBuffer, Optional[List[IRNode]]], str],
H A Dck_template.py89 def torch_type_to_ck(self, node: IRNode, ptr: str) -> str:
H A Drocm_template.py108 epilogue_nodes: Optional[List[IRNode]] = None,
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_template_kernel.py205 nodes: List[ir.IRNode],
262 epilogue_nodes: Optional[List[ir.IRNode]] = None,
337 Optional[List[ir.IRNode]],
H A Dcpp_template.py86 epilogue_nodes: Optional[List[ir.IRNode]] = None,
H A Dcpp_utils.py629 nodes: List[ir.IRNode],
647 def wrap_inner_fn_for_node(node: ir.IRNode):
H A Dcpp_gemm_template.py771 epilogue_nodes: Optional[List[ir.IRNode]] = None,
H A Dsimd.py739 def call_kernel(self, name: str, node: Optional[IRNode] = None) -> None:
H A Dtriton.py2811 def call_kernel(self, name: str, node: Optional[IRNode] = None):
/aosp_15_r20/prebuilts/go/linux-x86/src/cmd/compile/internal/devirtualize/
Dpgo_test.go74 func addEdge(caller, callee *pgoir.IRNode, offset int, weight int64) {
/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dflex_attention.py78 def maybe_realize(args: List[Optional[IRNode]]):
/aosp_15_r20/external/skia/src/sksl/ir/
H A DSkSLIRNode.h145 IRNode(Position position, int kind) in IRNode() function