Check the blocksize

491c0631 · rocking · 7a7d50ec · 491c0631
Commit 491c0631 authored Feb 13, 2023 by rocking
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

include/ck/tensor_operation/gpu/device/impl/device_normalization_impl.hpp ...r_operation/gpu/device/impl/device_normalization_impl.hpp +1 -0

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_normalization_impl.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_normalization_impl.hpp
@@ -50,6 +50,7 @@ struct DeviceNormalizationImpl : public DeviceNormalization<XDataType,
                                                            Rank,
                                                            NumReduceDim>
 {
+    static_assert(BlockSize == MThreadClusterSize * KThreadClusterSize);
    static_assert(
        ((GammaSrcVectorDim == 0 && MThreadSliceSize % GammaSrcVectorSize == 0) ||
         (GammaSrcVectorDim == 1 && KThreadSliceSize % GammaSrcVectorSize == 0)),