nixpkgs/pkgs/development/python-modules/pytorch
Benjamin Hipple fb5bb25c10 python3Packages.pytorch: fixup builds with MKL
- Pass `blas.provider` into `buildInputs`, so that CMake can find the actual
  `mkl` for inspection of its cmake files and headers.

- Add `USE_MKL` correctly when the blas provider is `mkl`.

- Use the MKLDNN and MKLDNN_CBLAS flags by default, since `mkldnn` is FOSS and
  always available..

- Remove a patch for MKL 2019, since we've moved to 2020.

- Add a pythonImportsCheck for "torch" as a basic sanity-check

- Removed some unused variables at the top of the file
2020-05-09 13:30:54 -04:00
..
default.nix python3Packages.pytorch: fixup builds with MKL 2020-05-09 13:30:54 -04:00