causal-conv1d 1.0.0__tar.gz → 1.0.1__tar.gz

{causal_conv1d-1.0.0 → causal_conv1d-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: causal_conv1d
-Version: 1.0.0
+Version: 1.0.1
 Summary: Causal depthwise conv1d in CUDA, with a PyTorch interface
 Home-page: https://github.com/Dao-AILab/causal-conv1d
 Author: Tri Dao

{causal_conv1d-1.0.0 → causal_conv1d-1.0.1}/causal_conv1d/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "1.0.0"
+__version__ = "1.0.1"
 
 from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update

{causal_conv1d-1.0.0 → causal_conv1d-1.0.1}/causal_conv1d.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: causal-conv1d
-Version: 1.0.0
+Version: 1.0.1
 Summary: Causal depthwise conv1d in CUDA, with a PyTorch interface
 Home-page: https://github.com/Dao-AILab/causal-conv1d
 Author: Tri Dao

{causal_conv1d-1.0.0 → causal_conv1d-1.0.1}/causal_conv1d.egg-info/requires.txt

@@ -1,3 +1,4 @@
 torch
 packaging
+buildtools
 ninja

{causal_conv1d-1.0.0 → causal_conv1d-1.0.1}/setup.py

@@ -3,6 +3,7 @@ import sys
 import warnings
 import os
 import re
+import shutil
 import ast
 from pathlib import Path
 from packaging.version import parse, Version
@@ -147,7 +148,7 @@ if not SKIP_CUDA_BUILD:
                     + cc_flag
                 ),
             },
-            include_dirs=[this_dir],
+            include_dirs=[Path(this_dir) / "csrc" / "causal_conv1d"],
         )
     )
 
@@ -216,7 +217,7 @@ class CachedWheelsCommand(_bdist_wheel):
 
             wheel_path = os.path.join(self.dist_dir, archive_basename + ".whl")
             print("Raw wheel path", wheel_path)
-            os.rename(wheel_filename, wheel_path)
+            shutil.move(wheel_filename, wheel_path)
         except urllib.error.HTTPError:
             print("Precompiled wheel not found. Building from source...")
             # If the wheel could not be downloaded, build from source
@@ -259,6 +260,7 @@ setup(
     install_requires=[
         "torch",
         "packaging",
+        "buildtools",
         "ninja",
     ],
 )