-
Notifications
You must be signed in to change notification settings - Fork 193
/
meta.yaml
114 lines (108 loc) · 4.51 KB
/
meta.yaml
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
106
107
108
109
110
111
112
113
114
# Copyright (c) 2018-2022, NVIDIA CORPORATION.
{% set version = environ.get('GIT_DESCRIBE_TAG', '0.0.0.dev').lstrip('v') + environ.get('VERSION_SUFFIX', '') %}
{% set cuda_version='.'.join(environ.get('RAPIDS_CUDA_VERSION', '11.5.2').split('.')[:2]) %}
{% set cuda_major=cuda_version.split('.')[0] %}
{% set cuda_spec = ">=" + cuda_major ~ ",<" + (cuda_major | int + 1) ~ ".0a0" %} # i.e. >=11,<12.0a0
package:
name: librmm-split
source:
git_url: ../../..
requirements:
build:
- cmake {{ cmake_version }}
- ninja
- {{ compiler('c') }}
- {{ compiler('cxx') }}
- {{ compiler('cuda') }} {{ cuda_version }}
- sysroot_{{ target_platform }} {{ sysroot_version }}
host:
- cudatoolkit {{ cuda_version }}.*
build:
script_env:
- PARALLEL_LEVEL
- CMAKE_GENERATOR
- CMAKE_C_COMPILER_LAUNCHER
- CMAKE_CXX_COMPILER_LAUNCHER
- CMAKE_CUDA_COMPILER_LAUNCHER
- SCCACHE_S3_KEY_PREFIX=librmm-aarch64 # [aarch64]
- SCCACHE_S3_KEY_PREFIX=librmm-linux64 # [linux64]
- SCCACHE_BUCKET
- SCCACHE_REGION
- SCCACHE_IDLE_TIMEOUT
- AWS_ACCESS_KEY_ID
- AWS_SECRET_ACCESS_KEY
outputs:
- name: librmm
version: {{ version }}
script: install_librmm.sh
build:
number: {{ GIT_DESCRIBE_NUMBER }}
string: cuda{{ cuda_major }}_{{ GIT_DESCRIBE_HASH }}_{{ GIT_DESCRIBE_NUMBER }}
run_exports:
- {{ pin_subpackage("librmm", max_pin="x.x") }}
ignore_run_exports_from:
- {{ compiler('cuda') }}
requirements:
build:
- cmake {{ cmake_version }}
run:
- cudatoolkit {{ cuda_spec }}
- spdlog {{ spdlog_version }}
test:
commands:
- test -f $PREFIX/include/rmm/thrust_rmm_allocator.h
- test -f $PREFIX/include/rmm/logger.hpp
- test -f $PREFIX/include/rmm/cuda_stream.hpp
- test -f $PREFIX/include/rmm/cuda_stream_view.hpp
- test -f $PREFIX/include/rmm/cuda_stream_pool.hpp
- test -f $PREFIX/include/rmm/device_uvector.hpp
- test -f $PREFIX/include/rmm/device_scalar.hpp
- test -f $PREFIX/include/rmm/device_buffer.hpp
- test -f $PREFIX/include/rmm/detail/aligned.hpp
- test -f $PREFIX/include/rmm/detail/error.hpp
- test -f $PREFIX/include/rmm/detail/exec_check_disable.hpp
- test -f $PREFIX/include/rmm/mr/device/detail/arena.hpp
- test -f $PREFIX/include/rmm/mr/device/detail/free_list.hpp
- test -f $PREFIX/include/rmm/mr/device/detail/coalescing_free_list.hpp
- test -f $PREFIX/include/rmm/mr/device/detail/fixed_size_free_list.hpp
- test -f $PREFIX/include/rmm/mr/device/detail/stream_ordered_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/arena_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/binning_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/cuda_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/device_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/fixed_size_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/limiting_resource_adaptor.hpp
- test -f $PREFIX/include/rmm/mr/device/logging_resource_adaptor.hpp
- test -f $PREFIX/include/rmm/mr/device/managed_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/owning_wrapper.hpp
- test -f $PREFIX/include/rmm/mr/device/per_device_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/pool_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/device/thread_safe_resource_adaptor.hpp
- test -f $PREFIX/include/rmm/mr/device/thrust_allocator_adaptor.hpp
- test -f $PREFIX/include/rmm/mr/host/host_memory_resource.hpp
- test -f $PREFIX/include/rmm/mr/host/new_delete_resource.hpp
- test -f $PREFIX/include/rmm/mr/host/pinned_memory_resource.hpp
about:
home: http://rapids.ai/
license: Apache-2.0
summary: librmm library
- name: librmm-tests
version: {{ version }}
script: install_librmm_tests.sh
build:
number: {{ GIT_DESCRIBE_NUMBER }}
string: cuda{{ cuda_major }}_{{ GIT_DESCRIBE_HASH }}_{{ GIT_DESCRIBE_NUMBER }}
ignore_run_exports_from:
- {{ compiler('cuda') }}
requirements:
build:
- cmake {{ cmake_version }}
run:
- cudatoolkit {{ cuda_spec }}
- {{ pin_subpackage('librmm', exact=True) }}
- gtest {{ gtest_version }}
- gmock {{ gtest_version }}
about:
home: http://rapids.ai/
license: Apache-2.0
summary: librmm test & benchmark executables