Skip to content

Commit bddf531

Browse files
committed
hotfix linter
Signed-off-by: ignaciosica <[email protected]>
1 parent 9007b97 commit bddf531

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vllm/platforms/cpu.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,9 @@ def supported_dtypes(self) -> list[torch.dtype]:
7777
return [torch.bfloat16, torch.float32]
7878
elif (self.get_cpu_architecture() == CpuArchEnum.ARM
7979
and sys.platform.startswith("darwin")):
80-
if (subprocess.check_output(["sysctl -n hw.optional.arm.FEAT_BF16"],
81-
shell=True).strip() == b"1"):
80+
if (subprocess.check_output(
81+
["sysctl -n hw.optional.arm.FEAT_BF16"],
82+
shell=True).strip() == b"1"):
8283
return [torch.bfloat16, torch.float16, torch.float32]
8384
return [torch.float16, torch.float32]
8485
# x86/aarch64 CPU has supported both bf16 and fp16 natively.

0 commit comments

Comments
 (0)