Commit d0b8de20 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python312Packages.torch: cleanup dependencies

parent 256db2e5

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

+20 −19
Original line number Diff line number Diff line
@@ -44,15 +44,18 @@ 


  # dependencies

  astunparse,

  fsspec,

  expecttest,

  filelock,

  fsspec,

  hypothesis,

  jinja2,

  networkx,

  sympy,

  numpy,

  packaging,

  psutil,

  pyyaml,

  cffi,

  click,

  requests,

  sympy,

  types-dataclasses,

  typing-extensions,

  # ROCm build and `torch.compile` requires `triton`

  tritonSupport ? (!stdenv.hostPlatform.isDarwin),
@@ -69,10 +72,6 @@ 
  _tritonEffective ? if cudaSupport then triton-cuda else triton,

  triton-cuda,



  # Unit tests

  hypothesis,

  psutil,



  # Disable MKLDNN on aarch64-darwin, it negatively impacts performance,

  # this is also what official pytorch build does

  mklDnnSupport ? !(stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isAarch64),
@@ -510,18 +509,20 @@ buildPythonPackage rec { 
  dependencies =

    [

      astunparse

      cffi

      click

      numpy

      pyyaml



      # From install_requires:

      fsspec

      expecttest

      filelock

      typing-extensions

      sympy

      networkx

      fsspec

      hypothesis

      jinja2

      networkx

      ninja

      packaging

      psutil

      pyyaml

      requests

      sympy

      types-dataclasses

      typing-extensions



      # the following are required for tensorboard support

      pillow