diff --git a/pyproject.toml b/pyproject.toml index c27c9d508..034c94070 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -17,8 +17,8 @@ classifiers = [ ] dependencies = [ "torch", - "tilelang==0.1.8", - "apache-tvm-ffi<=0.1.12", + "tilelang>=0.1.9,<0.1.10", + "apache-tvm-ffi<0.1.10", "quack-kernels>=0.3.4", "triton>=3.5.0", "ninja", diff --git a/setup.py b/setup.py index 29fc07977..bcd7b5e94 100755 --- a/setup.py +++ b/setup.py @@ -399,8 +399,8 @@ def run(self): "einops", "triton>=3.5.0", "transformers", - "tilelang==0.1.8", - "apache-tvm-ffi<=0.1.12", + "tilelang>=0.1.9,<0.1.10", + "apache-tvm-ffi<0.1.10", "quack-kernels>=0.3.4", # "causal_conv1d>=1.4.0", ],