diff --git a/pkgs/development/python-modules/causal-conv1d/default.nix b/pkgs/development/python-modules/causal-conv1d/default.nix index d4dc4cba2010..8902d04fc0e9 100644 --- a/pkgs/development/python-modules/causal-conv1d/default.nix +++ b/pkgs/development/python-modules/causal-conv1d/default.nix @@ -14,9 +14,6 @@ which, }: -assert cudaSupport -> !rocmSupport; -assert rocmSupport -> !cudaSupport; - buildPythonPackage rec { pname = "causal-conv1d"; version = "1.5.4"; @@ -87,7 +84,9 @@ buildPythonPackage rec { description = "Causal depthwise conv1d in CUDA with a PyTorch interface"; homepage = "https://github.com/Dao-AILab/causal-conv1d"; license = lib.licenses.bsd3; - # The package requires CUDA or ROCm - broken = !(cudaSupport || rocmSupport); + + # The package requires either CUDA or ROCm. + # It doesn't work without either, nor with both. + broken = cudaSupport != rocmSupport; }; }