catopy/meson.build at develop · vergelli/catopy · 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
project('cato', 'cpp', 'cuda',
  version : '0.1.0',
  license : 'MIT',
  meson_version : '>= 0.60.0',
  default_options : ['cpp_std=c++17', 'cuda_std=c++17']
)

# Encontrar Python
py_mod = import('python')
py3 = py_mod.find_installation('python3', required: true)

# Encontrar pybind11
pybind11_dep = dependency('pybind11', required: true)

# Encontrar spdlog
spdlog_dep = dependency('spdlog', required: true)

# Encontrar CUDA
cuda_dep = dependency('cuda', required: true)

# Configurar arquitectura CUDA
cuda_arch = '80'
add_project_arguments('-arch=compute_' + cuda_arch, language: 'cuda')
add_project_arguments('-arch=sm_' + cuda_arch, language: 'cuda')

# Incluir directorios
incdirs = include_directories(
  'include/core',
  'include/device',
  'include/device/wrappers',
  'include/host',
  'include/logger'
)

# Fuentes del proyecto
sources = [
  'src/host/bindings/bindings.cpp',
  'src/host/bindings/devices_bindings.cpp',
  'src/host/bindings/data_structures_bindings.cpp',
  'src/host/bindings/operations/vector_operations.cpp',  # NEW
  'src/logger/LoggingConfig.cpp',
  'src/device/utils/devices.cu',
  'src/device/memOps/wrappers/caTensorInitValueWrapper.cu',
  'src/device/memOps/wrappers/caTensorInitZeroWrapper.cu',
  'src/device/optimizers/BaseLaunchOptimizer.cu',
  'src/device/optimizers/VectorLaunchOptimizer.cu',
  'src/device/optimizers/MatrixLaunchOptimizer.cu',
  'src/device/optimizers/TensorLaunchOptimizer.cu',
  'src/device/wrappers/VectorOperationWrapper.cu',
  'src/device/kernels/vector_kernels.cu',
]

# Crear el módulo Python
cato_module = py3.extension_module('cato',
  sources,
  include_directories : incdirs,
  dependencies : [pybind11_dep, spdlog_dep, cuda_dep],
  install : true
)