Update tinygemm2.cpp

Signed-off-by: dongfengy <[email protected]>

Author: dongfengy

cpp/tensorrt_llm/thop/tinygemm2.cpp

@@ -31,7 +31,7 @@ namespace torch_ext
 torch::Tensor tinygemm2_forward(torch::Tensor input, torch::Tensor weight, torch::Tensor bias)
 {
     auto const smVersion = tensorrt_llm::common::getSMVersion();
-    TORCH_CHECK(smVersion == 90 || smVersion == 100, "tinygemm2 only supports SM90 and SM100.");
+    TORCH_CHECK(smVersion == 90 || smVersion == 100 || smVersion == 103, "tinygemm2 only supports SM90, SM100, and SM103.");
     TORCH_CHECK(input.dim() == 2, "input must be 2D");
     TORCH_CHECK(weight.dim() == 2, "weight must be 2D");
     TORCH_CHECK(bias.dim() == 1, "bias must be 1D");