NVIDIA/cutlass
1
0
Fork 0
mirror of https://github.com/NVIDIA/cutlass.git synced 2026-04-20 14:59:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

add requirements.txt

Browse Source
This commit is contained in:
Fung Xie
2025-11-25 21:03:11 -08:00
parent 2664cac685
commit 286781a1fb
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/python/CuTeDSL/cute/tvm_ffi/ampere_gemm_with_fake_tensor.py
View File

@@ -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

2
examples/python/CuTeDSL/cute/tvm_ffi/requirements.txt Normal file
View File

@@ -0,0 +1,2 @@
apache-tvm-ffi
torch-c-dlpack-ext