Commit ac43ec34 authored by Yaroslav Bolyukin's avatar Yaroslav Bolyukin
Browse files

torch: passthru rocmSupport flag

parent 12d167f5

pkgs/development/python-modules/torch/default.nix

+1 −1
Original line number Diff line number Diff line
@@ -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