amdgpu-device-libs 
Support library for the amdgpu target.
By default, the amdgpu target supports core
, but not std
.
alloc
is supported when a global allocator is specified.
amdgpu-device-libs
brings some std-like features to the amdgpu target:
print!()
andprintln!()
macros for printing on the host stdout- A global allocator to support
alloc
- A panic handler
- Access to more intrinsics and device-libs functions
All these features are enabled by default, but can be turned on selectively with default-features = false, features = […]
.
amdgpu-device-libs
works by linking to the ROCm device-libs and a pre-compiled helper library.
The libraries are linked from a ROCm installation.
To make sure the libraries are found, set the environment variable ROCM_PATH
or ROCM_DEVICE_LIB_PATH
(higher priority if it is set).
It looks for amdgcn/bitcode/*.bc
files in this path.
Usage
Create a new cargo library project and change it to compile a cdylib:
# Cargo.toml
# Force lto
[]
= true
[]
= true
[]
# Compile a cdylib
= ["cdylib"]
[]
# Used in build script to specify linker flags and link in device-libs
= { = "../../amdgpu-device-libs-build" }
[]
= { = "../../amdgpu-device-libs" }
Add extra flags in .cargo/config.toml
:
# .cargo/config.toml
[]
= "amdgcn-amd-amdhsa"
# Enable linker-plugin-lto and workarounds
# Either add -Ctarget-cpu=gfx<version> here or specify it in CARGO_BUILD_RUSTFLAGS='-Ctarget-cpu=gfx<version>'
= ["-Clinker-plugin-lto", "-Zemit-thin-lto=no"]
[]
= ["core", "alloc"]
And add a build.rs
build script that links to the required libraries:
// build.rs
Example
Minimal usage sample, see examples/println
for the full code.
extern crate alloc;
use Vec;
use *;
pub extern "gpu-kernel"