File tree 1 file changed +49
-0
lines changed 1 file changed +49
-0
lines changed Original file line number Diff line number Diff line change
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'
You can’t perform that action at this time.
0 commit comments