We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d94ca74 commit e8bef72Copy full SHA for e8bef72
src/ATen/native/xpu/sycl/LayerNormKernels.cpp
@@ -599,7 +599,7 @@ void _layer_norm_kernel(
599
beta.defined() ? can_vectorize(beta_data, alignment) : true;
600
601
if ((std::is_same_v<T, float> || std::is_same_v<T, at::Half> ||
602
- std::is_same_v<T, at::BFloat16>) &&
+ std::is_same_v<T, at::BFloat16>) &&
603
N <= static_cast<int64_t>(1ULL << std::numeric_limits<float>::digits) &&
604
N % num_vec_elems == 0 && can_vec_X && can_vec_Y && can_vec_gamma &&
605
can_vec_beta) {
0 commit comments