Fix pow triton 3.5 error (#1271)

Author: 0x45f

src/flag_gems/ops/pow.py

@@ -12,7 +12,7 @@
 @pointwise_dynamic(promotion_methods=[(0, 1, "BOOL_TO_LONG")])
 @triton.jit
 def pow_func(x, exponent):
-    return _pow(x.to(tl.float32), exponent)
+    return _pow(x.to(tl.float32), exponent.to(tl.float32))
 
 
 def pow_tensor_tensor(A, exponent):
@@ -28,7 +28,7 @@ def pow_tensor_tensor_(A, exponent):
 @pointwise_dynamic(is_tensor=[True, False], promotion_methods=[(0, 1, "BOOL_TO_LONG")])
 @triton.jit
 def pow_func_tensor_scalar(x, exponent):
-    return _pow(x.to(tl.float32), exponent)
+    return _pow(x.to(tl.float32), exponent.to(tl.float32))
 
 
 def pow_tensor_scalar(A, exponent):
@@ -44,7 +44,7 @@ def pow_tensor_scalar_(A, exponent):
 @pointwise_dynamic(is_tensor=[False, True], promotion_methods=[(0, 1, "BOOL_TO_LONG")])
 @triton.jit
 def pow_func_scalar_tensor(x, exponent):
-    return _pow(x.to(tl.float32), exponent)
+    return _pow(x.to(tl.float32), exponent.to(tl.float32))
 
 
 def pow_scalar(A, exponent):