Fix warning 186-D from nvcc (#2527)

2025-10-19 00:04:41 +08:00 · 2025-08-22 10:29:55 +09:00
parent f93f87c802
commit f6819a1f26
1 changed files with 4 additions and 2 deletions
--- a/mlx/backend/cuda/device/binary_ops.cuh
+++ b/mlx/backend/cuda/device/binary_ops.cuh
@@ -205,8 +205,10 @@ struct Power {
    if constexpr (cuda::std::is_integral_v<T>) {
      T res = 1;
      // Raising an integer to a negative power is undefined
-      if (exp < 0) {
-        return 0;
+      if constexpr (cuda::std::is_signed_v<T>) {
+        if (exp < 0) {
+          return 0;
+        }
      }
      while (exp) {
        if (exp & 1) {