Home
last modified time | relevance | path

Searched refs:HeuristicType (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/mlgo/
H A DMLGOParser.cpp531 HeuristicType heuristic_type(TokenStream &in, bool &valid, bool take = true) in heuristic_type()
536 return HeuristicType::GEMM_Type; in heuristic_type()
540 return HeuristicType::GEMM_Config_Native; in heuristic_type()
544 return HeuristicType::GEMM_Config_Reshaped_Only_RHS; in heuristic_type()
548 return HeuristicType::GEMM_Config_Reshaped; in heuristic_type()
552 … FAIL_WITH_MSG_DEFAULT(valid, HeuristicType::GEMM_Config_Reshaped, pos, "Expect heuristic type"); in heuristic_type()
556 void expect_heuristic_type(TokenStream &in, HeuristicType expected_ht, bool &valid) in expect_heuristic_type()
722 const HeuristicType t_heuristic_type = std::get<0>(t->index()); in heuristic_tree()
745 case HeuristicType::GEMM_Type: in heuristic_tree()
751 case HeuristicType::GEMM_Config_Native: in heuristic_tree()
[all …]
H A DUtils.cpp77 std::ostream &operator<<(std::ostream &os, HeuristicType ht) in operator <<()
81 case HeuristicType::GEMM_Type: in operator <<()
86 case HeuristicType::GEMM_Config_Reshaped_Only_RHS: in operator <<()
91 case HeuristicType::GEMM_Config_Reshaped: in operator <<()
133 HeuristicType ht; in operator <<()
H A DMLGOHeuristics.cpp67 auto index = std::make_tuple(HeuristicType::GEMM_Type, query.ip_target, query.data_type); in query_gemm_type()
85 …auto index = std::make_tuple(HeuristicType::GEMM_Config_Native, query.ip_target, query.data_t… in query_gemm_config_native()
103 …auto index = std::make_tuple(HeuristicType::GEMM_Config_Reshaped_Only_RHS, query.ip_target, q… in query_gemm_config_reshaped_only_rhs()
121 …auto index = std::make_tuple(HeuristicType::GEMM_Config_Reshaped, query.ip_target, query.data… in query_gemm_config_reshaped()
H A DHeuristicTree.h72 using Index = std::tuple<HeuristicType, std::string, DataType>;
119 … HeuristicTree(TreeID id, HeuristicType h_type, const std::string &ip_target, DataType data_type);
189 HeuristicType _heuristic_type; /**< Heuristic type */
H A DHeuristicTree.cpp96 : HeuristicTree(0, HeuristicType::GEMM_Type, "", DataType::F32) in HeuristicTree()
100 HeuristicTree::HeuristicTree(TreeID id, HeuristicType h_type, const std::string &ip_target, DataTyp… in HeuristicTree()
H A DCommon.h35 enum class HeuristicType enum
H A DUtils.h42 std::ostream &operator<<(std::ostream &os, HeuristicType ht);
/aosp_15_r20/external/pytorch/torch/_inductor/runtime/
H A Dtriton_heuristics.py29 HeuristicType,
266 and self.heuristic_type == HeuristicType.REDUCTION
786 self.heuristic_type == HeuristicType.TEMPLATE
787 or self.heuristic_type == HeuristicType.USER_AUTOTUNE
811 self.heuristic_type == HeuristicType.PERSISTENT_REDUCTION
1358 heuristic_type=HeuristicType.POINTWISE,
1375 heuristic_type=HeuristicType.POINTWISE,
1390 heuristic_type=HeuristicType.POINTWISE,
1407 heuristic_type=HeuristicType.POINTWISE,
1416 heuristic_type=HeuristicType.POINTWISE,
[all …]
H A Dhints.py78 class HeuristicType(Enum): class
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_triton_heuristics.py22 HeuristicType,
124 "heuristic_type": HeuristicType.POINTWISE,
H A Dtest_cuda_repro.py413 from torch._inductor.runtime.hints import HeuristicType, instance_descriptor
425 heuristic_type=HeuristicType.POINTWISE,