mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
b3504af56e
Part of: #123062 Ran lintrunner on: - `test/scripts` - `test/simulate_nccl_errors.py` - `test/test_ao_sparsity.py` - `test/test_autocast.py` - `test/test_binary_ufuncs.py` - `test/test_bundled_images.py` - `test/test_bundled_inputs.py` - `test/test_comparison_utils.py` - `test/test_compile_benchmark_util.py` - `test/test_complex.py` - `test/test_cpp_api_parity.py` - `test/test_cpp_extensions_aot.py` - `test/test_cpp_extensions_jit.py` - `test/test_cpp_extensions_open_device_registration.py` Detail: ```bash $ lintrunner -a --take UFMT --all-files ok No lint issues. Successfully applied all patches. ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/124137 Approved by: https://github.com/soulitzer
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
#!/usr/bin/env python3
|
|
# Owner(s): ["module: internals"]
|
|
|
|
import torch
|
|
from torch.testing._internal.common_utils import run_tests, TestCase
|
|
|
|
|
|
class TestComparisonUtils(TestCase):
|
|
def test_all_equal_no_assert(self):
|
|
t = torch.tensor([0.5])
|
|
torch._assert_tensor_metadata(t, [1], [1], torch.float)
|
|
|
|
def test_all_equal_no_assert_nones(self):
|
|
t = torch.tensor([0.5])
|
|
torch._assert_tensor_metadata(t, None, None, None)
|
|
|
|
def test_assert_dtype(self):
|
|
t = torch.tensor([0.5])
|
|
|
|
with self.assertRaises(RuntimeError):
|
|
torch._assert_tensor_metadata(t, None, None, torch.int32)
|
|
|
|
def test_assert_strides(self):
|
|
t = torch.tensor([0.5])
|
|
|
|
with self.assertRaises(RuntimeError):
|
|
torch._assert_tensor_metadata(t, None, [3], torch.float)
|
|
|
|
def test_assert_sizes(self):
|
|
t = torch.tensor([0.5])
|
|
|
|
with self.assertRaises(RuntimeError):
|
|
torch._assert_tensor_metadata(t, [3], [1], torch.float)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|