Skip to content

Commit

Permalink
Fix formatting check
Browse files Browse the repository at this point in the history
  • Loading branch information
loadams committed Mar 12, 2024
1 parent 16a8c18 commit 53ecd23
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/unit/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def bf16_required_version_check(accelerator_check=True):
cuda_version_available = CUDA_MAJOR >= 11
nccl_version_available = NCCL_MAJOR > 2 or (NCCL_MAJOR == 2 and NCCL_MINOR >= 10)
npu_available = get_accelerator().device_name() == 'npu'
hpu_available = get_Accelerator().device_name() == 'hpu'
hpu_available = get_accelerator().device_name() == 'hpu'

if torch_version_available and cuda_version_available and nccl_version_available and accelerator_pass:
return True
Expand Down

0 comments on commit 53ecd23

Please sign in to comment.