Ufmt and bash fix

Author: alugorey

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
 

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

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