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