Skip to content

Commit 1a95f10

Browse files
committed
init
1 parent c561d26 commit 1a95f10

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchao/experimental/ops/linear_8bit_act_xbit_weight/linear_8bit_act_xbit_weight.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,8 @@ void linear_operator(
203203
nc = tiling_params->nc;
204204
} else {
205205
auto params = LinearTilingParams::from_target_tiles_per_thread(
206-
m,
206+
// We process m sequentially, so m_step is the "m" for the purpose of computing tiling params
207+
m_step,
207208
m_step,
208209
n,
209210
n_step,

0 commit comments

Comments
 (0)