Searched defs:check_model_cuda (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ | ||
| H A D | test_combo_kernels.py | 32 check_model_cuda = check_model_cuda variable in ComboKernelTests |
| H A D | test_cuda_repro.py | 58 check_model_cuda = test_torchinductor.check_model_cuda variable |
| H A D | test_foreach.py | 89 check_model_cuda = check_model_cuda variable in ForeachTests |
| H A D | test_torchinductor.py | 650 check_model_cuda = check_model_gpu variable |