mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
PYFMT lint grandfathered files 1 (#154261)
lint: - test/test_fake_tensor.py - test/test_flop_counter.py - torch/_export/verifier.py with same rules as other files, it was a night mare for me to update tests in one of the skipped files with not being able to lint them locally like other files with lintrunner -a. note that those file do have active dev and not old not touched files. Pull Request resolved: https://github.com/pytorch/pytorch/pull/154261 Approved by: https://github.com/angelayi, https://github.com/Skylion007
This commit is contained in:
committed by
PyTorch MergeBot
parent
5677ab9aab
commit
43b2716e89
@ -3731,7 +3731,7 @@ class NcclProcessGroupWithDispatchedCollectivesTests(
|
||||
@parametrize("float8_dtype", [torch.float8_e4m3fn, torch.float8_e5m2])
|
||||
def test_allgather_float8(self, float8_dtype):
|
||||
device = torch.device(f"cuda:{self.rank:d}")
|
||||
if not sm_is_or_higher_than(device, 9, 0):
|
||||
if not sm_is_or_higher_than(device, 9, 0): # noqa: F821
|
||||
self.skipTest("FP8 reduction support begins with sm90 capable devices")
|
||||
store = dist.FileStore(self.file_name, self.world_size)
|
||||
dist.init_process_group(
|
||||
|
Reference in New Issue
Block a user