[Hardware][AMD][CI] Fix AMD CI image build (#46792)
Signed-off-by: Matthew Wong <Matthew.Wong2@amd.com>
Matt committed
Jun 26, 2026 at 00:05 UTC
1a4984520ed06560db66ae21bbb11362fe82d0bd
1 file changed
-1
csrc/libtorch_stable/layernorm_kernels.cu
-1
@@ -29,7 +29,6 @@ __global__ void rms_norm_kernel(
29
float variance = 0.0f;
30
const scalar_t* input_row;
31
const scalar_t* weight_row;
32
- int64_t weight_row_off = 0;
32
if constexpr (NUM_DIMS == 2) {
33
// 2D for layernorm normal case [batch_size, hidden]
34
input_row = input + blockIdx.x * input_stride_d2;