From 14c179c2c952cd92879b7535ea01c1fd2dfeba2b Mon Sep 17 00:00:00 2001 From: Wei Feng Date: Sun, 22 Sep 2024 19:59:10 -0700 Subject: [PATCH] [float8] fix typo in bitwise_identical unit test Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: --- test/float8/test_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/float8/test_base.py b/test/float8/test_base.py index 2a875c44d..db66a206e 100644 --- a/test/float8/test_base.py +++ b/test/float8/test_base.py @@ -53,7 +53,7 @@ is_cuda_8_9 = torch.cuda.is_available() and torch.cuda.get_device_capability() >= (8, 9) def bitwise_identical(a: Float8Tensor, b: Float8Tensor) -> bool: - assert torch.all(a._data == b._data).item(), "scales are not identical" + assert torch.all(a._scale == b._scale).item(), "scales are not identical" assert torch.all(a._data == b._data).item(), "data is not identical" return True