meta.yaml

# 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