v4.4 tag release update. (#3032)

examples/python/CuTeDSL/cute/ffi/jit_argument.py

@@ -240,10 +240,11 @@ import os
 import subprocess
 import shutil
 import tempfile
-import torch
 
 
 def run_test(tmpdir=None, cmake_args="", cleanup=True):
+    import torch
+
     try:
         current_dir = os.path.dirname(os.path.abspath(__file__))
 

examples/python/CuTeDSL/cute/torch_fake_tensor.py

@@ -27,8 +27,6 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 
-import torch
-
 import cutlass.cute as cute
 from cutlass.cute.runtime import from_dlpack
 
@@ -66,6 +64,7 @@ def print_tensor(t: cute.Tensor):
 
 
 def run():
+    import torch
     from torch._subclasses.fake_tensor import FakeTensorMode
 
     shape = (3, 4)

examples/python/CuTeDSL/cute/tvm_ffi/aot_use_in_cpp_bundle.cpp

@@ -1,6 +1,6 @@
 // clang-format off
 /*
- * SPDX-FileCopyrightText: Copyright (c) 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-FileCopyrightText: Copyright (c) 2023 - 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
  * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
  *
  * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual

examples/python/CuTeDSL/cute/tvm_ffi/aot_use_in_cpp_bundle.sh

@@ -28,7 +28,7 @@
 
 #!/bin/bash
 # Set up library paths for runtime
-export LD_LIBRARY_PATH=$(python3 -m cutlass.cute.export.aot_config --libdir):$(tvm-ffi-config --libdir)
+export LD_LIBRARY_PATH=$(python3 -m cutlass.cute.export.aot_config --libdir):$(tvm-ffi-config --libdir):$LD_LIBRARY_PATH
 
 CUDA_HOME=/usr/local/cuda
 SOURCE_FILE="$(dirname "$0")/aot_use_in_cpp_bundle.cpp"