We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a0ff0db commit 819c9b8Copy full SHA for 819c9b8
cpp/tensorrt_llm/thop/fp8RowwiseGemm.cpp
@@ -71,7 +71,7 @@ torch::Tensor fp8_rowwise_gemm_launch(torch::Tensor const& mat1, torch::Tensor c
71
auto const k = mat1.sizes()[1];
72
73
static_assert(std::is_same<OutputType, half>::value || std::is_same<OutputType, __nv_bfloat16>::value,
74
- "Output type must be half, or bfloat16");
+ "Output type must be half or bfloat16");
75
static constexpr auto outType
76
= std::is_same<OutputType, half>::value ? at::ScalarType::Half : at::ScalarType::BFloat16;
77
at::Tensor out;
0 commit comments