Fix comment

lshpku · lshpku · commit 3ff52f0e88fb · 2025-05-28T11:36:45.000Z
diff --git a/slm/model_zoo/gpt-3/external_ops/fused_quanted_ops/fused_transpose_split_quant.cu b/slm/model_zoo/gpt-3/external_ops/fused_quanted_ops/fused_transpose_split_quant.cu
@@ -175,8 +175,12 @@ __global__ void __launch_bounds__(1024, 2) FusedTransposeSplitQuantKernel(
  *
  * Inputs:
  *   X     : [SUM(M_1...M_N), K], bfloat16
+ *
+ * Attrs:
  *   tokens_per_expert
- *         : python list of value [M_1, M_2, ..., M_N]
+ *         : list of value [M_1, M_2, ..., M_N]
+ *   pow_2_scales
+ *         : bool that indicates whether to use power-of-2 scaling
  *
  * Outputs:
  *   out   : [K * M_1 + K * M_2 + ... + K * M_N]

Original file line number	Diff line number	Diff line change
`@@ -175,8 +175,12 @@ __global__ void __launch_bounds__(1024, 2) FusedTransposeSplitQuantKernel(`
`175`	`175`	`*`
`176`	`176`	`* Inputs:`
`177`	`177`	`* X : [SUM(M_1...M_N), K], bfloat16`
	`178`	`+ *`
	`179`	`+ * Attrs:`
`178`	`180`	`* tokens_per_expert`
`179`		`- * : python list of value [M_1, M_2, ..., M_N]`
	`181`	`+ * : list of value [M_1, M_2, ..., M_N]`
	`182`	`+ * pow_2_scales`
	`183`	`+ * : bool that indicates whether to use power-of-2 scaling`
`180`	`184`	`*`
`181`	`185`	`* Outputs:`
`182`	`186`	`* out : [K * M_1 + K * M_2 + ... + K * M_N]`