| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_logging.py | 13 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_minifier_isolate.py | 16 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_group_batch_fusion.py | 23 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_inductor_freezing.py | 38 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_cudagraph_trees.py | 53 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_debug_utils.py | 13 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_minifier.py | 9 requires_cuda = unittest.skipUnless(torch.cuda.is_available(), "requires cuda") variable
|
| H A D | test_structured_trace.py | 28 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_backends.py | 18 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_logging.py | 28 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_activation_checkpointing.py | 33 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_subclasses.py | 144 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_higher_order_ops.py | 37 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | test_repros.py | 61 requires_cuda = unittest.skipUnless(torch.cuda.is_available(), "requires cuda") variable
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | triton_utils.py | 9 requires_cuda = unittest.skipUnless(HAS_CUDA, "requires cuda") variable
|
| H A D | common_utils.py | 1458 requires_cuda = unittest.skipUnless(torch.cuda.is_available(), "Requires CUDA") variable
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_converter.py | 20 requires_cuda = unittest.skipUnless(torch.cuda.is_available(), "requires cuda") variable
|