causal-conv1d 1.2.0.post2__tar.gz → 1.2.2.post1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (13) hide show
  1. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/PKG-INFO +1 -1
  2. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d/__init__.py +1 -1
  3. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d.egg-info/PKG-INFO +1 -1
  4. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/setup.py +8 -0
  5. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/AUTHORS +0 -0
  6. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/LICENSE +0 -0
  7. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/README.md +0 -0
  8. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d/causal_conv1d_interface.py +0 -0
  9. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d.egg-info/SOURCES.txt +0 -0
  10. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d.egg-info/dependency_links.txt +0 -0
  11. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d.egg-info/requires.txt +0 -0
  12. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d.egg-info/top_level.txt +0 -0
  13. {causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/setup.cfg +0 -0
{causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: causal_conv1d
3
- Version: 1.2.0.post2
3
+ Version: 1.2.2.post1
4
4
  Summary: Causal depthwise conv1d in CUDA, with a PyTorch interface
5
5
  Home-page: https://github.com/Dao-AILab/causal-conv1d
6
6
  Author: Tri Dao
{causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d/__init__.py RENAMED
@@ -1,3 +1,3 @@
1
- __version__ = "1.2.0.post2"
1
+ __version__ = "1.2.2.post1"
2
2
 
3
3
  from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update
{causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/causal_conv1d.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: causal-conv1d
3
- Version: 1.2.0.post2
3
+ Version: 1.2.2.post1
4
4
  Summary: Causal depthwise conv1d in CUDA, with a PyTorch interface
5
5
  Home-page: https://github.com/Dao-AILab/causal-conv1d
6
6
  Author: Tri Dao
{causal_conv1d-1.2.0.post2 → causal_conv1d-1.2.2.post1}/setup.py RENAMED
@@ -101,10 +101,18 @@ if not SKIP_CUDA_BUILD:
101
101
  "Note: make sure nvcc has a supported version by running nvcc -V."
102
102
  )
103
103
 
104
+ cc_flag.append("-gencode")
105
+ cc_flag.append("arch=compute_53,code=sm_53")
106
+ cc_flag.append("-gencode")
107
+ cc_flag.append("arch=compute_62,code=sm_62")
104
108
  cc_flag.append("-gencode")
105
109
  cc_flag.append("arch=compute_70,code=sm_70")
106
110
  cc_flag.append("-gencode")
111
+ cc_flag.append("arch=compute_72,code=sm_72")
112
+ cc_flag.append("-gencode")
107
113
  cc_flag.append("arch=compute_80,code=sm_80")
114
+ cc_flag.append("-gencode")
115
+ cc_flag.append("arch=compute_87,code=sm_87")
108
116
  if bare_metal_version >= Version("11.8"):
109
117
  cc_flag.append("-gencode")
110
118
  cc_flag.append("arch=compute_90,code=sm_90")