Add jinja comments to grad_indice_weights kernel

avbokovoy · avbokovoy · commit 585300ded18a · 2025-11-06T09:33:04.000Z
diff --git a/fbgemm_gpu/codegen/training/backward/embedding_backward_split_indice_weights_template.cu b/fbgemm_gpu/codegen/training/backward/embedding_backward_split_indice_weights_template.cu
@@ -333,9 +333,9 @@ __global__ __launch_bounds__(kForwardMaxThreads) void
                     }
                 }
             }
-            {%- endif %}
+            {%- endif %}{#-/* if not ssd and not dense and not use_vec_blocking and not vbe */#}
             for (; j < kWarpSize && l_start + j < L; ++j) {
-            {%- else %} // if is_rocm
+            {%- else %}{#-/* if is_rocm*/#}
             for (auto j = 0; j < kWarpSize && l_start + j < L; ++j) {
             {%- endif %} // if is_rocm
                 const auto offset_idx_j = shfl_sync(offset_idx, j);

Original file line number	Diff line number	Diff line change
`@@ -333,9 +333,9 @@ __global__ __launch_bounds__(kForwardMaxThreads) void`
`333`	`333`	`}`
`334`	`334`	`}`
`335`	`335`	`}`
`336`		`- {%- endif %}`
	`336`	`+ {%- endif %}{#-/* if not ssd and not dense and not use_vec_blocking and not vbe */#}`
`337`	`337`	`for (; j < kWarpSize && l_start + j < L; ++j) {`
`338`		`- {%- else %} // if is_rocm`
	`338`	`+ {%- else %}{#-/* if is_rocm*/#}`
`339`	`339`	`for (auto j = 0; j < kWarpSize && l_start + j < L; ++j) {`
`340`	`340`	`{%- endif %} // if is_rocm`
`341`	`341`	`const auto offset_idx_j = shfl_sync(offset_idx, j);`