fix the compare in PD_MEA_CHECK_OVERFLOW (#52300)

155018ee · Danyang Zhang · GitHub · 1faa06f0 · 155018ee
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/phi/kernels/fusion/cutlass/memory_efficient_attention_utils.h ...kernels/fusion/cutlass/memory_efficient_attention_utils.h +1 -1

未找到文件。
--- a/paddle/phi/kernels/fusion/cutlass/memory_efficient_attention_utils.h
+++ b/paddle/phi/kernels/fusion/cutlass/memory_efficient_attention_utils.h
@@ -67,7 +67,7 @@ inline int64_t GetMemoryEfficientBiasStrideB(const phi::DDim &bias_dims,
    auto __src = (__VA_ARGS__);                                              \
    using __SrcType = decltype(&__src);                                      \
    using __DstType = typename std::remove_reference<decltype(__dst)>::type; \
-    if (__src < std::numeric_limits<__DstType>::max()) {                     \
+    if (__src > std::numeric_limits<__DstType>::max()) {                     \
      PADDLE_THROW(                                                          \
          phi::errors::InvalidArgument(#__dst " exceeds maximum value."));   \
    }                                                                        \