Skip to content

Commit

Permalink
adding easyconfigs: ml_dtypes-0.3.2-gfbf-2023a.eb and patches: ml_dty…
Browse files Browse the repository at this point in the history
…pes-0.3.2_EigenAvx512.patch
  • Loading branch information
casparl committed Jun 3, 2024
1 parent 19f73c7 commit 7dccb4b
Show file tree
Hide file tree
Showing 2 changed files with 1,270 additions and 0 deletions.
51 changes: 51 additions & 0 deletions easybuild/easyconfigs/m/ml_dtypes/ml_dtypes-0.3.2-gfbf-2023a.eb
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
# Thomas Hoffmann, EMBL Heidelberg, [email protected], 2024/02
easyblock = 'PythonBundle'

name = 'ml_dtypes'
version = '0.3.2'

homepage = 'https://github.com/jax-ml/ml_dtypes'
description = """
ml_dtypes is a stand-alone implementation of several NumPy dtype extensions used
in machine learning libraries, including:
bfloat16: an alternative to the standard float16 format
float8_*: several experimental 8-bit floating point representations including:
float8_e4m3b11fnuz
float8_e4m3fn
float8_e4m3fnuz
float8_e5m2
float8_e5m2fnuz
"""

toolchain = {'name': 'gfbf', 'version': '2023a'}

dependencies = [
('Python', '3.11.3'),
('SciPy-bundle', '2023.07'),
]


use_pip = True

default_easyblock = 'PythonPackage'

exts_list = [
('opt_einsum', '3.3.0', {
'checksums': ['59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549'],
}),
('etils', '1.6.0', {
'checksums': ['c635fbd02a79fed4ad76825d31306b581d22b40671721daa8bc279cf6333e48a'],
}),
(name, version, {
'patches': [('ml_dtypes-0.3.2_EigenAvx512.patch', 1)],
'checksums': [
{'ml_dtypes-0.3.2.tar.gz': '533059bc5f1764fac071ef54598db358c167c51a718f68f5bb55e3dee79d2967'},
{'ml_dtypes-0.3.2_EigenAvx512.patch': '197b05b0b7f611749824369f026099f6a172f9e8eab6ebb6504a16573746c892'},
],
}),
]

sanity_pip_check = True

moduleclass = 'tools'
Loading

0 comments on commit 7dccb4b

Please sign in to comment.