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

   1 { lib
   2 , stdenv
   3 , anytree
   4 , buildPythonPackage
   5 , cached-property
   6 , cgen
   7 , click
   8 , codepy
   9 , distributed
  10 , fetchFromGitHub
  11 , gcc
  12 , llvmPackages
  13 , matplotlib
  14 , multidict
  15 , nbval
  16 , psutil
  17 , py-cpuinfo
  18 , pyrevolve
  19 , pytest-xdist
  20 , pytestCheckHook
  21 , pythonOlder
  22 , pythonRelaxDepsHook
  23 , scipy
  24 , sympy
  25 }:
  26
  27 buildPythonPackage rec {
  28   pname = "devito";
  29   version = "4.8.3";
  30   format = "setuptools";
  31
  32   disabled = pythonOlder "3.7";
  33
  34   src = fetchFromGitHub {
  35     owner = "devitocodes";
  36     repo = "devito";
  37     rev = "refs/tags/v${version}";
  38     hash = "sha256-g9rRJF1JrZ6+s3tj4RZHuGOjt5LJjtK9I5CJmq4CJL4=";
  39   };
  40
  41   pythonRemoveDeps = [
  42     "codecov"
  43     "flake8"
  44     "pytest-runner"
  45     "pytest-cov"
  46   ];
  47
  48   pythonRelaxDeps = true;
  49
  50   nativeBuildInputs = [
  51     pythonRelaxDepsHook
  52   ];
  53
  54   propagatedBuildInputs = [
  55     anytree
  56     cached-property
  57     cgen
  58     click
  59     codepy
  60     distributed
  61     nbval
  62     multidict
  63     psutil
  64     py-cpuinfo
  65     pyrevolve
  66     scipy
  67     sympy
  68   ] ++ lib.optionals stdenv.cc.isClang [
  69     llvmPackages.openmp
  70   ];
  71
  72   nativeCheckInputs = [
  73     gcc
  74     matplotlib
  75     pytest-xdist
  76     pytestCheckHook
  77   ];
  78
  79   pytestFlagsArray = [ "-x" ];
  80
  81   # I've had to disable the following tests since they fail while using nix-build, but they do pass
  82   # outside the build. They mostly related to the usage of MPI in a sandboxed environment.
  83   disabledTests = [
  84     "test_assign_parallel"
  85     "test_cache_blocking_structure_distributed"
  86     "test_codegen_quality0"
  87     "test_coefficients_w_xreplace"
  88     "test_docstrings"
  89     "test_docstrings[finite_differences.coefficients]"
  90     "test_gs_parallel"
  91     "test_if_halo_mpi"
  92     "test_if_parallel"
  93     "test_init_omp_env_w_mpi"
  94     "test_loop_bounds_forward"
  95     "test_mpi_nocomms"
  96     "test_mpi"
  97     "test_index_derivative"
  98     "test_new_distributor"
  99     "test_setupWOverQ"
 100     "test_shortcuts"
 101     "test_subdomainset_mpi"
 102   ];
 103
 104   disabledTestPaths = [
 105     "tests/test_pickle.py"
 106     "tests/test_benchmark.py"
 107     "tests/test_mpi.py"
 108     "tests/test_autotuner.py"
 109     "tests/test_data.py"
 110     "tests/test_dse.py"
 111     "tests/test_gradient.py"
 112   ];
 113
 114   pythonImportsCheck = [
 115     "devito"
 116   ];
 117
 118   meta = with lib; {
 119     description = "Code generation framework for automated finite difference computation";
 120     homepage = "https://www.devitoproject.org/";
 121     changelog = "https://github.com/devitocodes/devito/releases/tag/v${version}";
 122     license = licenses.mit;
 123     maintainers = with maintainers; [ atila ];
 124   };
 125 }