FilesExpand file tree

 3.8.0

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

106 lines (91 loc) · 4.41 KB

 3.8.0

/

setup.py

Top

File metadata and controls

• Code
• Blame

106 lines (91 loc) · 4.41 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

#!/usr/bin/env python

#######################################################

# Copyright (c) 2015, ArrayFire

# All rights reserved.

#

# This file is distributed under 3-clause BSD license.

# The complete license agreement can be obtained at:

# http://arrayfire.com/licenses/BSD-3-Clause

########################################################

import os

import re

# package can be distributed with arrayfire binaries or

# just with python wrapper files, the AF_BUILD_LOCAL

# environment var determines whether to build the arrayfire

# binaries locally rather than searching in a system install

AF_BUILD_LOCAL_LIBS = os.environ.get('AF_BUILD_LOCAL_LIBS')

print(f'AF_BUILD_LOCAL_LIBS={AF_BUILD_LOCAL_LIBS}')

if AF_BUILD_LOCAL_LIBS:

print('Proceeding to build ArrayFire libraries')

else:

print('Skipping binaries installation, only python files will be installed')

AF_BUILD_CPU = os.environ.get('AF_BUILD_CPU')

AF_BUILD_CPU = 1 if AF_BUILD_CPU is None else int(AF_BUILD_CPU)

AF_BUILD_CPU_CMAKE_STR = '-DAF_BUILD_CPU:BOOL=ON' if (AF_BUILD_CPU == 1) else '-DAF_BUILD_CPU:BOOL=OFF'

AF_BUILD_CUDA = os.environ.get('AF_BUILD_CUDA')

AF_BUILD_CUDA = 1 if AF_BUILD_CUDA is None else int(AF_BUILD_CUDA)

AF_BUILD_CUDA_CMAKE_STR = '-DAF_BUILD_CUDA:BOOL=ON' if (AF_BUILD_CUDA == 1) else '-DAF_BUILD_CUDA:BOOL=OFF'

AF_BUILD_OPENCL = os.environ.get('AF_BUILD_OPENCL')

AF_BUILD_OPENCL = 1 if AF_BUILD_OPENCL is None else int(AF_BUILD_OPENCL)

AF_BUILD_OPENCL_CMAKE_STR = '-DAF_BUILD_OPENCL:BOOL=ON' if (AF_BUILD_OPENCL == 1) else '-DAF_BUILD_OPENCL:BOOL=OFF'

AF_BUILD_UNIFIED = os.environ.get('AF_BUILD_UNIFIED')

AF_BUILD_UNIFIED = 1 if AF_BUILD_UNIFIED is None else int(AF_BUILD_UNIFIED)

AF_BUILD_UNIFIED_CMAKE_STR = '-DAF_BUILD_UNIFIED:BOOL=ON' if (AF_BUILD_UNIFIED == 1) else '-DAF_BUILD_UNIFIED:BOOL=OFF'

if AF_BUILD_LOCAL_LIBS:

# invoke cmake and build arrayfire libraries to install locally in package

from skbuild import setup

def filter_af_files(cmake_manifest):

cmake_manifest = list(filter(lambda name: not (name.endswith('.h')

or name.endswith('.cpp')

or name.endswith('.hpp')

or name.endswith('.cmake')

or name.endswith('jpg')

or name.endswith('png')

or name.endswith('libaf.so') #avoids duplicates due to symlinks

or re.match('.*libaf\.so\.3\..*', name) is not None

or name.endswith('libafcpu.so')

or re.match('.*libafcpu\.so\.3\..*', name) is not None

or name.endswith('libafcuda.so')

or re.match('.*libafcuda\.so\.3\..*', name) is not None

or name.endswith('libafopencl.so')

or re.match('.*libafopencl\.so\.3\..*', name) is not None

or name.endswith('libforge.so')

or re.match('.*libforge\.so\.1\..*', name) is not None

or 'examples' in name), cmake_manifest))

return cmake_manifest

print('Building CMAKE with following configurable variables: ')

print(AF_BUILD_CPU_CMAKE_STR)

print(AF_BUILD_CUDA_CMAKE_STR)

print(AF_BUILD_OPENCL_CMAKE_STR)

print(AF_BUILD_UNIFIED_CMAKE_STR)

setup(

packages=['arrayfire'],

cmake_install_dir='',

cmake_process_manifest_hook=filter_af_files,

include_package_data=False,

cmake_args=[AF_BUILD_CPU_CMAKE_STR,

AF_BUILD_CUDA_CMAKE_STR,

AF_BUILD_OPENCL_CMAKE_STR,

AF_BUILD_UNIFIED_CMAKE_STR,

# todo: pass additional args from environ

'-DCMAKE_BUILD_TYPE:STRING="RelWithDebInfo"',

'-DFG_USE_STATIC_CPPFLAGS:BOOL=OFF',

'-DFG_WITH_FREEIMAGE:BOOL=OFF',

'-DCUDA_architecture_build_targets:STRING=All',

'-DAF_BUILD_DOCS:BOOL=OFF',

'-DAF_BUILD_EXAMPLES:BOOL=OFF',

'-DAF_INSTALL_STANDALONE:BOOL=ON',

'-DAF_WITH_IMAGEIO:BOOL=ON',

'-DAF_WITH_LOGGING:BOOL=ON',

'-DBUILD_TESTING:BOOL=OFF',

'-DAF_BUILD_FORGE:BOOL=ON',

'-DAF_INSTALL_LIB_DIR:STRING=arrayfire',

'-DAF_INSTALL_BIN_DIR:STRING=arrayfire',

'-DFG_INSTALL_LIB_DIR:STRING=arrayfire',

'-DAF_WITH_STATIC_MKL=ON',

]

)

else:

# ignores local arrayfire libraries, will search system instead

from setuptools import setup

setup()