torch: passthru rocmSupport flag

2024-03-02 10:02:30 +01:00
parent 12d167f551
commit ac43ec34d9
1 changed files with 1 additions and 1 deletions
@@ -492,7 +492,7 @@ in buildPythonPackage rec {
  requiredSystemFeatures = [ "big-parallel" ];

  passthru = {
-    inherit cudaSupport cudaPackages;
+    inherit cudaSupport cudaPackages rocmSupport rocmPackages;
    # At least for 1.10.2 `torch.fft` is unavailable unless BLAS provider is MKL. This attribute allows for easy detection of its availability.
    blasProvider = blas.provider;
    # To help debug when a package is broken due to CUDA support