fix: update global variable syntax since its not support from torch 3.2.0

Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>

Author: kmehant

plugins/fused-ops-and-kernels/src/fms_acceleration_foak/kernels/unsloth/cross_entropy_loss.py

@@ -187,7 +187,7 @@ def _cross_entropy_backward(
 pass
 
 
-MAX_FUSED_SIZE: tl.constexpr = 65536 # 2**16
+MAX_FUSED_SIZE = tl.constexpr(65536) # 2**16
 
 class Fast_CrossEntropyLoss(torch.autograd.Function):
     @staticmethod

plugins/fused-ops-and-kernels/src/fms_acceleration_foak/kernels/unsloth/rope_embedding.py

@@ -17,7 +17,7 @@
 import torch
 from .utils import calculate_settings
 
-ROPE_GROUP_SIZE: tl.constexpr = 4
+ROPE_GROUP_SIZE = tl.constexpr(4)
 
 @triton.heuristics({"BACKWARD_PASS": lambda args: args["BACKWARD_PASS"],})
 @triton.jit