Unverified Commit 752e1c1a authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.torchcodec: inherit cudaSupport from torch (#498205)

parents 59afd7e7 fee78689

maintainers/maintainer-list.nix

+4 −1
Original line number Diff line number Diff line
@@ -4352,7 +4352,10 @@ 
    github = "caniko";

    githubId = 29519599;

    name = "Can H. Tartanoglu";

    keys = [ { fingerprint = "DF95 1EC0 9B8F 8094 C616  5589 1D63 6EDE 97DC 0280"; } ];

    keys = [

      { fingerprint = "DF95 1EC0 9B8F 8094 C616  5589 1D63 6EDE 97DC 0280"; }

      { fingerprint = "818D 507F 1E62 139F 8A17  EAA6 4623 DEA0 6FDA CFE1"; }

    ];

  };

  canndrew = {

    email = "shum@canndrew.org";

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

+6 −4
Original line number Diff line number Diff line 
{

  lib,

  stdenv,

  config,

  buildPythonPackage,

  fetchFromGitHub,
@@ -10,7 +9,6 @@ 


  # buildInputs

  ffmpeg,

  cudaPackages,



  # build-system

  cmake,
@@ -21,7 +19,8 @@ 
  pytestCheckHook,

  torchvision,



  cudaSupport ? config.cudaSupport,

  cudaSupport ? torch.cudaSupport,

  cudaPackages,

}:



buildPythonPackage (finalAttrs: {
@@ -167,6 +166,9 @@ buildPythonPackage (finalAttrs: { 
    homepage = "https://github.com/meta-pytorch/torchcodec";

    changelog = "https://github.com/meta-pytorch/torchcodec/releases/tag/${finalAttrs.src.tag}";

    license = lib.licenses.bsd3;

    maintainers = with lib.maintainers; [ GaetanLepage ];

    maintainers = with lib.maintainers; [

      GaetanLepage

      caniko

    ];

  };

})