Unverified Commit 3b932f5c authored by Nick Cao's avatar Nick Cao Committed by GitHub
Browse files

python3Packages.helion: 0.3.2 -> 1.0.0 (#507373)

parents 359a67cd adf584d0

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

+12 −13
Original line number Diff line number Diff line
@@ -10,13 +10,12 @@ 


  # dependencies

  filecheck,

  numpy,

  psutil,

  rich,

  scikit-learn,

  torch,

  tqdm,

  triton,

  typing-extensions,

  torch,



  # tests

  pytestCheckHook,
@@ -26,14 +25,14 @@ 


buildPythonPackage (finalAttrs: {

  pname = "helion";

  version = "0.3.2";

  version = "1.0.0";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "pytorch";

    repo = "helion";

    tag = "v${finalAttrs.version}";

    hash = "sha256-MR8fo6MhMSAXlBsx//ODIprXPXFhMy6K5Mno9BnZGHc=";

    hash = "sha256-3Iam+1FUg9I9kKmkQBZp9/FTZpEjf4Ba+cKRo5eLEzw=";

  };



  build-system = [
@@ -43,13 +42,15 @@ buildPythonPackage (finalAttrs: { 


  dependencies = [

    filecheck

    numpy

    psutil

    rich

    scikit-learn

    torch

    tqdm

    triton

    typing-extensions



    # torch is not listed as a dependency, but is actually required at import time

    # https://github.com/pytorch/helion/blob/v1.0.0/helion/_compat.py#L13

    torch

  ];



  pythonImportsCheck = [ "helion" ];
@@ -67,12 +68,10 @@ buildPythonPackage (finalAttrs: { 


  # Tests require GPU access

  doCheck = false;

  passthru.gpuChecks = {

    pytest = helion.overridePythonAttrs {

  passthru.gpuCheck = helion.overridePythonAttrs {

    doCheck = true;

    requiredSystemFeatures = [ "cuda" ];

  };

  };



  meta = {

    description = "Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate";