added supported restrict in device_elementwise_normalization_impl.hpp

0fc1858e · guangzlu · 03fc4e35 · 0fc1858e
Commit 0fc1858e authored Feb 09, 2023 by guangzlu
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

include/ck/tensor_operation/gpu/device/impl/device_elementwise_normalization_impl.hpp ...gpu/device/impl/device_elementwise_normalization_impl.hpp +5 -0

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_elementwise_normalization_impl.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_elementwise_normalization_impl.hpp
@@ -533,6 +533,11 @@ struct DeviceElementwiseNormalizationImpl
                return (false);
        }

+        if(p_arg_->x_lds_size_ >= 65536)
+        {
+            return (false);
+        }
+
        return true;
    };