causal-conv1d 1.5.0.post5__tar.gz → 1.5.0.post8__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.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: causal_conv1d
3
- Version: 1.5.0.post5
3
+ Version: 1.5.0.post8
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
@@ -1,3 +1,3 @@
1
- __version__ = "1.5.0.post5"
1
+ __version__ = "1.5.0.post8"
2
2
 
3
3
  from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: causal-conv1d
3
- Version: 1.5.0.post5
3
+ Version: 1.5.0.post8
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
@@ -267,9 +267,9 @@ def get_wheel_url():
267
267
  # We're using the CUDA version used to build torch, not the one currently installed
268
268
  # _, cuda_version_raw = get_cuda_bare_metal_version(CUDA_HOME)
269
269
  torch_cuda_version = parse(torch.version.cuda)
270
- # For CUDA 11, we only compile for CUDA 11.8, and for CUDA 12 we only compile for CUDA 12.4
270
+ # For CUDA 11, we only compile for CUDA 11.8, and for CUDA 12 we only compile for CUDA 12.3
271
271
  # to save CI time. Minor versions should be compatible.
272
- torch_cuda_version = parse("11.8") if torch_cuda_version.major == 11 else parse("12.4")
272
+ torch_cuda_version = parse("11.8") if torch_cuda_version.major == 11 else parse("12.3")
273
273
  cuda_version = f"{torch_cuda_version.major}"
274
274
 
275
275
  gpu_compute_version = hip_version if HIP_BUILD else cuda_version