From 286781a1fb5365d281b4a4598115ad8f89cbf97b Mon Sep 17 00:00:00 2001 From: Fung Xie Date: Tue, 25 Nov 2025 21:03:11 -0800 Subject: [PATCH] add requirements.txt --- .../CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py | 3 +-- examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt | 2 ++ 2 files changed, 3 insertions(+), 2 deletions(-) create mode 100644 examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt diff --git a/examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py b/examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py index f7624da3..68e8f81a 100644 --- a/examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py +++ b/examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py @@ -29,7 +29,6 @@ import sys import os import torch -import time import cutlass import cutlass.cute as cute @@ -43,7 +42,7 @@ This example shows how to compile CuTe JIT function with fake tensors then run i if __name__ == "__main__": # Add the current directory to sys.path current_dir = os.path.dirname(os.path.abspath(__file__)) - sys.path.insert(0, os.path.join(current_dir, "..")) + sys.path.insert(0, os.path.join(current_dir, "..", "..")) from ampere.tensorop_gemm import TensorOpGemm diff --git a/examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt b/examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt new file mode 100644 index 00000000..8a3f9dcd --- /dev/null +++ b/examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt @@ -0,0 +1,2 @@ +apache-tvm-ffi +torch-c-dlpack-ext \ No newline at end of file