Unverified Commit 877fcd28 authored by Pol Dellaiera's avatar Pol Dellaiera Committed by GitHub
Browse files

python312Packages.pytorch-bench: init at unstable-2024-07-18 (#352827)

parents 2fd5238d f803fa80

pkgs/development/python-modules/pytorch-bench/default.nix

0 → 100644
+49 −0
Original line number Diff line number Diff line 
{

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  setuptools,

  colorama,

  matplotlib,

  numpy,

  pynvml,

  torch,

  torchprofile,

}:



buildPythonPackage {

  pname = "pytorch-bench";

  version = "unstable-2024-07-18";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "MaximeGloesener";

    repo = "torch-benchmark";

    rev = "405a3fc2d147b43b4c1f7edb7aca0a60ba343ac5";

    hash = "sha256-KU3dAf97A6lkMNTKRay23BMFQfn1ReAFNaJ0kG2RfnA=";

  };



  build-system = [

    setuptools

  ];



  dependencies = [

    colorama

    matplotlib

    numpy

    pynvml

    torch

    torchprofile

  ];



  pythonImportsCheck = [

    "pytorch_bench"

  ];



  meta = {

    description = "Benchmarking tool for torch";

    homepage = "https://github.com/MaximeGloesener/torch-benchmark";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ drupol ];

  };

}

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

0 → 100644
+48 −0
Original line number Diff line number Diff line 
{

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  setuptools,

  numpy,

  torch,

  torchvision,

}:



buildPythonPackage rec {

  pname = "torchprofile";

  version = "0.0.3";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "zhijian-liu";

    repo = "torchprofile";

    rev = "refs/tags/v${version}";

    hash = "sha256-6vxZHQwBjKpy288wcANdJ9gmvIOZloLv+iN76TtqYAI=";

  };



  build-system = [

    setuptools

  ];



  pythonRelaxDeps = [

    "torchvision"

  ];



  dependencies = [

    numpy

    torch

    torchvision

  ];



  pythonImportsCheck = [

    "torchprofile"

  ];



  meta = {

    changelog = "https://github.com/zhijian-liu/torchprofile/releases/tag/v${version}";

    description = "General and accurate MACs / FLOPs profiler for PyTorch models";

    homepage = "https://github.com/zhijian-liu/torchprofile";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ drupol ];

  };

}

pkgs/top-level/python-packages.nix

+4 −0
Original line number Diff line number Diff line
@@ -13093,6 +13093,8 @@ self: super: with self; { 














  pytools = callPackage ../development/python-modules/pytools { };



























  pytorch-bench = callPackage ../development/python-modules/pytorch-bench { };



























  pytorch-lightning = callPackage ../development/python-modules/pytorch-lightning { };





























  pytorch-metric-learning = callPackage ../development/python-modules/pytorch-metric-learning { };










@@ -15925,6 +15927,8 @@ self: super: with self; {













  # Used by streamlit, 2021-01-29















  tornado_5 = callPackage ../development/python-modules/tornado/5.nix { };



























  torchprofile = callPackage ../development/python-modules/torchprofile { };



























  torpy = callPackage ../development/python-modules/torpy { };





























  torrent-parser = callPackage ../development/python-modules/torrent-parser { };