Ufmt and bash fix

pytorch-labs · Jun 18, 2024 · 08a1029 · 08a1029
1 parent f32a4a4
commit 08a1029
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/float8_experimental/float8_utils.py b/float8_experimental/float8_utils.py
@@ -6,11 +6,11 @@
 
 from typing import Literal, Tuple
 
+import float8_experimental.config as config
+
 import torch
 import torch.distributed as dist
 
-import float8_experimental.config as config
-
 # Helpful visualizer for debugging (only supports fp32):
 # https://www.h-schmidt.net/FloatConverter/IEEE754.html
 

diff --git a/test/test_dtensor.py b/test/test_dtensor.py
@@ -25,7 +25,7 @@
     Float8RowwiseParallel,
     PrepareFloat8ModuleInput,
 )
-from float8_experimental.float8_utils import tensor_to_scale, e4m3_dtype
+from float8_experimental.float8_utils import e4m3_dtype, tensor_to_scale
 from torch.distributed._tensor import distribute_tensor, DTensor, Replicate, Shard
 from torch.distributed.device_mesh import DeviceMesh, init_device_mesh
 from torch.distributed.tensor.parallel import parallelize_module

diff --git a/test/test_everything.sh b/test/test_everything.sh
@@ -2,7 +2,7 @@
 
 # terminate script on first error
 set -e
-IS_ROCM=$(rocm-smi --version)
+IS_ROCM=$(rocm-smi --version || true)
 
 pytest test/test_base.py
 pytest test/test_sam.py