Skip to content

Commit a60ca92

Browse files
committed
Update tinygemm2.cpp
Signed-off-by: dongfengy <[email protected]>
1 parent 7f753d9 commit a60ca92

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cpp/tensorrt_llm/thop/tinygemm2.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,8 @@ namespace torch_ext
3131
torch::Tensor tinygemm2_forward(torch::Tensor input, torch::Tensor weight, torch::Tensor bias)
3232
{
3333
auto const smVersion = tensorrt_llm::common::getSMVersion();
34-
TORCH_CHECK(smVersion == 90 || smVersion == 100 || smVersion == 103, "tinygemm2 only supports SM90, SM100, and SM103.");
34+
TORCH_CHECK(
35+
smVersion == 90 || smVersion == 100 || smVersion == 103, "tinygemm2 only supports SM90, SM100, and SM103.");
3536
TORCH_CHECK(input.dim() == 2, "input must be 2D");
3637
TORCH_CHECK(weight.dim() == 2, "weight must be 2D");
3738
TORCH_CHECK(bias.dim() == 1, "bias must be 1D");

0 commit comments

Comments
 (0)