Skip to content

Commit 6beebc9

Browse files
committed
#487 prepare for swap-deps update
1 parent 27d4e82 commit 6beebc9

File tree

1 file changed

+49
-0
lines changed

1 file changed

+49
-0
lines changed
Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,49 @@
1+
easyblock = 'PythonBundle'
2+
3+
name = 'Horovod'
4+
version = '0.28.1'
5+
local_tf_version = '2.13.0'
6+
versionsuffix = '-TensorFlow-%s' % local_tf_version
7+
8+
homepage = 'https://github.com/uber/horovod'
9+
description = "Horovod is a distributed training framework for TensorFlow."
10+
11+
toolchain = {'name': 'foss', 'version': '2023a'}
12+
13+
builddependencies = [
14+
('CMake', '3.26.3'),
15+
]
16+
dependencies = [
17+
('Python', '3.11.3'),
18+
('PyYAML', '6.0'),
19+
('TensorFlow', local_tf_version),
20+
]
21+
22+
use_pip = True
23+
sanity_pip_check = True
24+
25+
local_preinstallopts = 'module swap protobuf/3.21.9-GCCcore-12.3.0 && HOROVOD_WITH_MPI=1 '
26+
local_preinstallopts += 'HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1 '
27+
28+
exts_list = [
29+
('cloudpickle', '2.2.1', {
30+
'checksums': ['d89684b8de9e34a2a43b3460fbca07d09d6e25ce858df4d5a44240403b6178f5'],
31+
}),
32+
('horovod', version, {
33+
'preinstallopts': local_preinstallopts,
34+
'patches': ['Horovod-0.28.1_support_flatbuffers_2.0.6.patch'],
35+
'checksums': [
36+
'92a43f5a94c43907a56805bad15f19700c62ffc83b7ca483f9e104e229f67ef0',
37+
'9696ffb3b2bad1d6dd5a9f37bc58078ca7c585f933bcbec037036ad9fc0b297d',
38+
],
39+
}),
40+
]
41+
42+
sanity_check_paths = {
43+
'files': ['bin/horovodrun'],
44+
'dirs': ['lib/python%(pyshortver)s/site-packages'],
45+
}
46+
47+
sanity_check_commands = ["horovodrun --help"]
48+
49+
moduleclass = 'tools'

0 commit comments

Comments
 (0)