pytorch/torch/testing/_internal
Nikita Shulga 6a985d8b2e Make inductor_utils.requires_gpu accept MPS (#145156)
Not yet ready to setp HAS_GPU to true, but can unskip tests that require GPU
(Noticed while running test_mps_basics.py that `test_scalar_cpu_tensor_arg` is getting skipped)

- Replace `GPU_TYPE` with `self.device` in `test_custom_op_fixed_layout_sequential`, `test_inductor_layout_optimization_input_mutations`, `test_mutable_custom_op_fixed_layout2`  otherwise they GPU tests are just running for _cpu suffixes.
- Tweak `test_tmp_not_defined_issue3` to work correctly on CPU, by defining `test_device` and `test_device_0`
- UnXFail `test_mutable_custom_op_fixed_layout2_dynamic_shapes` as it should just work on CPU
- Add `skip_if_no_triton` decorator and decorate `test_reduction_config_limit` with it, as it does not need CPU nor GPU, but rather a triton backend.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145156
Approved by: https://github.com/dcci, https://github.com/Skylion007, https://github.com/jansel
2025-02-06 01:14:36 +00:00
..
codegen
data
distributed Make regex error catching compatible with Python 3.12+. (#145945) 2025-02-05 00:57:36 +00:00
generated
opinfo
optests [opcheck] Improve error reporting; allow atol/rtol overrides (#146488) 2025-02-05 21:25:06 +00:00
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py
common_cuda.py [inductor triton] Disable incorrect TF32 usage on CUDA capability < 8 (#145684) 2025-01-28 22:01:08 +00:00
common_device_type.py
common_dist_composable.py
common_distributed.py
common_dtype.py [BE]: Enable ruff SLOT checks (#146276) 2025-02-04 19:18:23 +00:00
common_fsdp.py
common_jit.py
common_methods_invocations.py Bump nn.functional.conv3d tolerances for test_comprehensive (#135719) 2025-02-05 18:34:12 +00:00
common_mkldnn.py
common_modules.py
common_nn.py
common_optimizers.py
common_pruning.py
common_quantization.py
common_quantized.py
common_subclass.py
common_utils.py
composite_compliance.py
custom_op_db.py
custom_tensor.py
dist_utils.py
dynamo_test_failures.py
fake_config_module.py config: Support str env variables (#145980) 2025-01-30 00:13:02 +00:00
fake_config_module2.py
fake_config_module3.py
hop_db.py Barebones flat_apply HOP (#146060) 2025-02-01 16:17:48 +00:00
hypothesis_utils.py
inductor_utils.py Make inductor_utils.requires_gpu accept MPS (#145156) 2025-02-06 01:14:36 +00:00
jit_metaprogramming_utils.py [BE]: Enable ruff SLOT checks (#146276) 2025-02-04 19:18:23 +00:00
jit_utils.py
logging_tensor.py
logging_utils.py
quantization_torch_package_models.py
static_module.py
subclasses.py
torchbind_impls.py
triton_utils.py
two_tensor.py