Skip to content

Commit eba28af

Browse files
authored
Comment tweaks
Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>
1 parent 80543a6 commit eba28af

2 files changed

Lines changed: 1 addition & 2 deletions

File tree

transformer_engine/common/include/transformer_engine/hadamard_transform.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ void nvte_group_hadamard_transform_cast_fusion_columnwise(
115115
* \param[in] split_sections Array specifying splits in dimension 0 for each output tensor.
116116
* \param[in] num_tensors Number of output tensors, must be > 0.
117117
* \param[in] quant_config Quantization configuration.
118-
* \param[in] quant_workspace Quantization workspace, a device buffer that can be opaque
118+
* \param[in] quant_workspace Workspace buffer. Must be at least 4 bytes.
119119
* \param[in] stream CUDA stream used for the operation.
120120
*/
121121
void nvte_group_hadamard_transform_cast_fusion(const NVTETensor input, NVTETensor* outputs,

transformer_engine/pytorch/csrc/extensions/cast.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -878,7 +878,6 @@ void split_quantize_nvfp4_impl_with_rht_helper(const TensorWrapper &input,
878878
auto nvte_tile_scheduler_workspace =
879879
makeTransformerEngineTensor(tile_scheduler_workspace_torch);
880880
// call the fully-fused grouped kernel for rowwise quantization & colwise RHT quantization transpose
881-
// Note that the workspace field can be opaque, this will leave room for furture extensions
882881
nvte_group_hadamard_transform_cast_fusion(
883882
input.data(), reinterpret_cast<NVTETensor *>(nvte_tensor_output_list.data()),
884883
rht_matrix_nvte.data(), split_sections.data(), num_tensors, quant_config_list[0],

0 commit comments

Comments
 (0)