Skip to content

Commit e1cfca5

Browse files
committed
Update torchao/csrc/cpu/aten_kernels/float8_linear.cpp
1 parent 3bacf14 commit e1cfca5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torchao/csrc/cpu/aten_kernels/float8_linear.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -453,7 +453,9 @@ void _float8_linear_impl(
453453
TORCH_CHECK(weight.size(3) == block_n, "Float8 linear: unexpected weight shape");
454454
int64_t N = Nc * block_n;
455455
TORCH_CHECK(K == Kc * block_k, "Float8 linear: weight and input shapes mismatch");
456-
auto [parallel_on_M, block_m, Mc, Mc_parallel] = get_m_blocking(M);
456+
bool parallel_on_M;
457+
int64_t block_m, Mc, Mc_parallel;
458+
std::tie(parallel_on_M, block_m, Mc, Mc_parallel) = get_m_blocking(M);
457459
int64_t num_parallel_blocks = Mc_parallel * Nc;
458460

459461
// scales shape = [Nc, G, block_n]

0 commit comments

Comments
 (0)