Skip to content

Commit

Permalink
Fix build error - error: function "torchao::marlin_24_gemm" has alrea…
Browse files Browse the repository at this point in the history
…dy been defined (previous definition at line 83) (#863)

fix compile error
  • Loading branch information
kshitij12345 authored Sep 10, 2024
1 parent 3ac2ab8 commit d2226a4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torchao/csrc/cuda/sparse_marlin/marlin_kernel_nm.cu
Original file line number Diff line number Diff line change
Expand Up @@ -852,8 +852,6 @@ __global__ void Marlin_24(
}
}

#endif

#define CALL_IF_2_4(NUM_BITS, THREAD_M_BLOCKS, THREAD_N_BLOCKS, \
THREAD_K_BLOCKS, GROUP_BLOCKS) \
else if (num_bits == NUM_BITS && thread_m_blocks == THREAD_M_BLOCKS && \
Expand Down Expand Up @@ -1119,6 +1117,8 @@ torch::Tensor marlin_24_gemm(torch::Tensor& a, torch::Tensor& b_q_weight,
return c;
}

#endif

TORCH_LIBRARY_IMPL(torchao, CUDA, m) {
m.impl("torchao::marlin_24_gemm", &marlin_24_gemm);
}
Expand Down

0 comments on commit d2226a4

Please sign in to comment.