v4.4 tag release update. (#3032)

examples/python/CuTeDSL/ampere/call_from_jit.py

@@ -60,11 +60,8 @@ import os
 import sys
 from typing import Type, Tuple
 
-import torch
-
 import cutlass
 import cutlass.cute as cute
-from cutlass.torch import dtype as torch_dtype
 from cutlass.cute.runtime import make_ptr
 
 if __name__ == "__main__":
@@ -205,6 +202,9 @@ def tensor_op_gemm_wrapper(
 
 
 def run_tensor_op_gemm_wrapper(mnkl: Tuple[int, int, int, int]):
+    import torch
+    from cutlass.torch import dtype as torch_dtype
+
     print("\nRunning TensorOpGemm test with:")
     print(f"Tensor dimensions: {mnkl}")