Register XPU Dispatch LayerNormKernel

min-jean-cho · min-jean-cho · commit d9b5fd057e95 · 2025-02-27T01:29:22.000-08:00
diff --git a/src/ATen/native/xpu/LayerNorm.cpp b/src/ATen/native/xpu/LayerNorm.cpp
@@ -177,6 +177,8 @@ ::std::tuple<at::Tensor, at::Tensor, at::Tensor> layer_norm_backward_xpu(
       grad_bias,
       grad_input_mask);
 }
+
+REGISTER_XPU_DISPATCH(LayerNormKernel, &xpu::layer_norm_kernel);
 } // namespace native
 
 } // namespace at

Original file line number	Diff line number	Diff line change
`@@ -177,6 +177,8 @@ ::std::tuple<at::Tensor, at::Tensor, at::Tensor> layer_norm_backward_xpu(`
`177`	`177`	`grad_bias,`
`178`	`178`	`grad_input_mask);`
`179`	`179`	`}`
	`180`	`+`
	`181`	`+REGISTER_XPU_DISPATCH(LayerNormKernel, &xpu::layer_norm_kernel);`
`180`	`182`	`} // namespace native`
`181`	`183`
`182`	`184`	`} // namespace at`