update variable naming and comments for rocm (#5299)

2025-04-12 01:15:05 -05:00
parent 2eb55770f9
commit 3c9740d200
6 changed files with 14 additions and 10 deletions
--- a/sgl-kernel/tests/test_per_tensor_quant_fp8.py
+++ b/sgl-kernel/tests/test_per_tensor_quant_fp8.py
@@ -7,8 +7,8 @@ from sgl_kernel import sgl_per_tensor_quant_fp8

 from sglang.srt.utils import is_hip

-is_hip_ = is_hip()
-fp8_type_ = torch.float8_e4m3fnuz if is_hip_ else torch.float8_e4m3fn
+_is_hip = is_hip()
+fp8_type_ = torch.float8_e4m3fnuz if _is_hip else torch.float8_e4m3fn


 def sglang_scaled_fp8_quant(