hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix bug in py-dgl installation, add unit tests (#15617)

Browse source
This commit is contained in:
Adam J. Stewart 2020-03-21 12:52:01 -04:00 committed by GitHub
parent 4b771bc521
commit 7bde39d883
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 36 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
var/spack/repos/builtin/packages/py-dgl/package.py
View file

@ -47,6 +47,7 @@ class PyDgl(CMakePackage):
build_directory = 'build' build_directory = 'build'
# https://docs.dgl.ai/install/index.html#install-from-source
def cmake_args(self): def cmake_args(self):
args = [] args = []
@ -83,12 +84,45 @@ def cmake_args(self):
def install(self, spec, prefix): def install(self, spec, prefix):
with working_dir('python'): with working_dir('python'):
import os
print('Current dir:', os.getcwd())
setup_py('install', '--prefix=' + prefix, setup_py('install', '--prefix=' + prefix,
'--single-version-externally-managed', '--root=/') '--single-version-externally-managed', '--root=/')
# Work around installation bug: https://github.com/dmlc/dgl/issues/1379
install_tree(prefix.dgl, prefix.lib)
def setup_run_environment(self, env): def setup_run_environment(self, env):
# https://docs.dgl.ai/install/backend.html # https://docs.dgl.ai/install/backend.html
backend = self.spec.variants['backend'].value backend = self.spec.variants['backend'].value
env.set('DGLBACKEND', backend) env.set('DGLBACKEND', backend)
@property
def import_modules(self):
modules = [
'dgl', 'dgl.nn', 'dgl.runtime', 'dgl.backend', 'dgl.function',
'dgl.contrib', 'dgl._ffi', 'dgl.data', 'dgl.runtime.ir',
'dgl.backend.numpy', 'dgl.contrib.sampling', 'dgl._ffi._cy2',
'dgl._ffi._cy3', 'dgl._ffi._ctypes',
]
if 'backend=pytorch' in self.spec:
modules.extend([
'dgl.nn.pytorch', 'dgl.nn.pytorch.conv', 'dgl.backend.pytorch'
])
elif 'backend=mxnet' in self.spec:
modules.extend([
'dgl.nn.mxnet', 'dgl.nn.mxnet.conv', 'dgl.backend.mxnet'
])
elif 'backend=tensorflow' in self.spec:
modules.extend([
'dgl.nn.tensorflow', 'dgl.nn.tensorflow.conv',
'dgl.backend.tensorflow'
])
return modules
@run_after('install')
@on_package_attributes(run_tests=True)
def import_module_test(self):
with working_dir('spack-test', create=True):
for module in self.import_modules:
python('-c', 'import {0}'.format(module))