| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_select_algorithm.py | 8 import torch._inductor.select_algorithm as select_algorithm namespace
31 patch.object(select_algorithm, "VERIFY", dict(atol=1e-4, rtol=1e-4)),
32 patch.object(select_algorithm.AlgorithmSelectorCache, "lookup", skip_cache),
86 @patch.object(select_algorithm, "VERIFY", dict(atol=5e-2, rtol=5e-2))
356 caller = select_algorithm.TritonTemplateCaller(
|
| H A D | test_cpu_select_algorithm.py | 15 import torch._inductor.select_algorithm as select_algorithm namespace
61 if isinstance(choice, select_algorithm.ExternKernelCaller):
77 patch.object(select_algorithm, "VERIFY", dict(atol=1e-4, rtol=1e-4)),
78 patch.object(select_algorithm.AlgorithmSelectorCache, "lookup", skip_cache),
99 with patch.object(select_algorithm, "VERIFY", dict(atol=atol, rtol=rtol)):
1165 select_algorithm, "VERIFY", dict(atol=atol, rtol=rtol)
1302 select_algorithm, "VERIFY", dict(atol=atol, rtol=rtol)
1350 with patch.object(select_algorithm, "VERIFY", dict(atol=atol, rtol=rtol)):
|
| H A D | test_cutlass_backend.py | 15 from torch._inductor.select_algorithm import NoValidChoicesError
|
| H A D | test_max_autotune.py | 20 from torch._inductor.select_algorithm import (
|
| /aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
| H A D | unpack_mixed_mm.py | 5 from ..select_algorithm import autotune_select_algorithm, TritonTemplate
|
| H A D | mm_plus_mm.py | 7 from ..select_algorithm import (
|
| H A D | bmm.py | 7 from ..select_algorithm import (
|
| H A D | mm_scaled.py | 11 from ..select_algorithm import (
|
| H A D | mm.py | 26 from ..select_algorithm import (
473 from torch._inductor.select_algorithm import realize_inputs
|
| H A D | flex_decoding.py | 14 from ..select_algorithm import autotune_select_algorithm, TritonTemplate
|
| H A D | mm_common.py | 10 from torch._inductor.select_algorithm import realize_inputs
|
| H A D | conv.py | 18 from ..select_algorithm import (
|
| H A D | flex_attention.py | 29 from ..select_algorithm import autotune_select_algorithm, realize_inputs, TritonTemplate
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | quantized_lowerings.py | 11 from .select_algorithm import (
|
| H A D | select_algorithm.py | 1580 if isinstance(choice, torch._inductor.select_algorithm.ExternKernelCaller):
1584 choice, torch._inductor.select_algorithm.TritonTemplateCaller
|
| H A D | autotune_process.py | 46 from torch._inductor.select_algorithm import TritonTemplateCaller
|
| H A D | mkldnn_lowerings.py | 23 from .select_algorithm import (
|
| H A D | ir.py | 3677 assert isinstance(caller, torch._inductor.select_algorithm.TritonTemplateCaller)
3688 assert isinstance(caller, torch._inductor.select_algorithm.TritonTemplateCaller)
|
| /aosp_15_r20/external/pytorch/torch/_inductor/autoheuristic/ |
| H A D | autoheuristic.py | 275 from torch._inductor.select_algorithm import (
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_template_kernel.py | 14 from ..select_algorithm import PartialRender
|
| H A D | cpp_gemm_template.py | 15 from ..select_algorithm import DataProcessorTemplateWrapper
|
| H A D | common.py | 1813 from ..select_algorithm import TritonTemplateKernel
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | b2b_gemm.py | 27 from ..select_algorithm import (
|
| /aosp_15_r20/external/AFLplusplus/src/ |
| H A D | afl-fuzz-one.c | 34 static int select_algorithm(afl_state_t *afl, u32 max_algorithm) { in select_algorithm() function
5091 switch (r = (select_algorithm(afl, r_max))) { in mopt_common_fuzzing()
|