Home
last modified time | relevance | path

Searched refs:check_model_gpu (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/test/inductor/
Dtest_xpu_basic.py23 from inductor.test_torchinductor import check_model_gpu, TestCase
33 common = check_model_gpu
Dtest_torchinductor_opinfo.py50 from .test_torchinductor import check_model, check_model_gpu
52 from test_torchinductor import check_model, check_model_gpu
564 check_model_gpu = check_model_gpu variable in TestInductorOpInfo
750 self.check_model_gpu(
Dtest_binary_folding.py19 from inductor.test_torchinductor import check_model, check_model_gpu, copy_tests
242 common = check_model_gpu
Dtest_compiled_optimizers.py271 from .test_torchinductor import check_model, check_model_gpu
273 from test_torchinductor import check_model, check_model_gpu
600 check_model_gpu = check_model_gpu variable in CompiledOptimizerTests
Dtest_torchinductor_dynamic_shapes.py44 check_model_gpu,
105 common = check_model_gpu
Dtest_torchinductor.py570 def check_model_gpu( function
650 check_model_cuda = check_model_gpu
10553 common = check_model_gpu