Skip to content

Commit 819c9b8

Browse files
committed
refine
Signed-off-by: Dylan Chen <[email protected]>
1 parent a0ff0db commit 819c9b8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cpp/tensorrt_llm/thop/fp8RowwiseGemm.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ torch::Tensor fp8_rowwise_gemm_launch(torch::Tensor const& mat1, torch::Tensor c
7171
auto const k = mat1.sizes()[1];
7272

7373
static_assert(std::is_same<OutputType, half>::value || std::is_same<OutputType, __nv_bfloat16>::value,
74-
"Output type must be half, or bfloat16");
74+
"Output type must be half or bfloat16");
7575
static constexpr auto outType
7676
= std::is_same<OutputType, half>::value ? at::ScalarType::Half : at::ScalarType::BFloat16;
7777
at::Tensor out;

0 commit comments

Comments
 (0)