Skip to content

Commit 324718f

Browse files
committed
address precommit
1 parent 4392e26 commit 324718f

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

benchmarks/cutlass_benchmarks/w8a8_benchmarks.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -119,13 +119,12 @@ def bench_fp8(
119119
def ceil_div(x: int, y: int) -> int:
120120
return (x + y - 1) // y
121121

122-
block_scale_a = torch.rand((m, ceil_div(k, 128)),
123-
device="cuda",
124-
dtype=torch.float32)
125-
block_scale_b = torch.rand(ceil_div(k, 128),
126-
ceil_div(n, 128),
127-
device="cuda",
128-
dtype=torch.float32)
122+
block_scale_a = torch.rand(
123+
(m, ceil_div(k, 128)), device="cuda", dtype=torch.float32
124+
)
125+
block_scale_b = torch.rand(
126+
ceil_div(k, 128), ceil_div(n, 128), device="cuda", dtype=torch.float32
127+
)
129128
block_scale_a_M_major = block_scale_a.t().contiguous().t()
130129
block_scale_b_K_major = block_scale_b.t().contiguous().t()
131130
bias = torch.zeros((n,), device="cuda", dtype=torch.bfloat16)

0 commit comments

Comments
 (0)