d3m-arrayfire-primitives/setup.py at master · arrayfire/d3m-arrayfire-primitives · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import os
import re
from setuptools import setup, find_packages
from setuptools.command.develop import develop
from setuptools.command.install import install
from subprocess import check_call

PACKAGE_NAME = 'af_primitives'

def read_package_variable(key):
    """Read the value of a variable from the package without importing."""
    module_path = os.path.join(PACKAGE_NAME, '__init__.py')
    with open(module_path) as module:
        for line in module:
            parts = line.strip().split(' ')
            if parts and parts[0] == key:
                return parts[-1].strip("'")
    raise KeyError("'{0}' not found in '{1}'".format(key, module_path))


def install_arrayfire():
    check_call('mkdir -p /opt/arrayfire'.split())
    #check_call('wget -P /tmp https://arrayfire.s3.amazonaws.com/3.7.2/ArrayFire-v3.7.2_Linux_x86_64.sh'.split())
    #check_call('bash /tmp/ArrayFire-v3.7.2_Linux_x86_64.sh --skip-license --prefix=/opt/arrayfire'.split())
    # check_call('echo \'export LD_PRELOAD=/opt/arrayfire/lib64/libafcuda.so\' >> ~/.bashrc'.split())
    # check_call('echo \'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/arrayfire/lib64\' >> ~/.bashrc'.split())
    check_call('export LD_PRELOAD=/opt/arrayfire/lib64/libafcuda.so'.split())
    check_call('export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/arrayfire/lib64'.split())


class PostDevelopStep(develop):
    def run(self):
        install_arrayfire()
        develop.run(self)


class PostInstallStep(install):
    def run(self):
        #install_arrayfire()
        install.run(self)


setup(
    name=PACKAGE_NAME,
    version=read_package_variable('__version__'),
    description='ArrayFire Primitives',
    author=read_package_variable('__author__'),
    packages=find_packages(exclude=['contrib', 'docs', 'tests*']),
    install_requires=[
        'arrayfire==3.7.20200213',
        'd3m==2020.5.18'
    ],
    url='https://github.com/arrayfire/d3m-arrayfire-primitives',
    keywords='d3m_primitive',
    entry_points={
        'd3m.primitives': [
            'classification.logistic_regression.ArrayFire = af_primitives.af_LogisticRegression:af_LogisticRegression',
            'classification.k_neighbors.ArrayFire = af_primitives.af_KNeigborsClassifier:af_KNeighborsClassifier',
            'data_cleaning.imputer.ArrayFire = af_primitives.afSKImputer:SKImputer',
            'data_cleaning.string_imputer.ArrayFire = af_primitives.afSKStringImputer:SKStringImputer',
            'classification.mlp.ArrayFire = af_primitives.afSKMLPClassifier:SKMLPClassifier',
            'regression.mlp.ArrayFire = af_primitives.afSKMLPRegressor:SKMLPRegressor',
        ],
    },
    cmdclass={
        'develop': PostDevelopStep,
        'install': PostInstallStep,
    },
)