1#[cfg(any(
2 feature = "cuda-12",
3 feature = "cuda-13",
4 feature = "rocm-7",
5 feature = "tpu",
6 feature = "neuron",
7 feature = "metal",
8))]
9use std::path::PathBuf;
10
11#[cfg(feature = "cuda-12")]
12pub fn pjrt_cuda_12_plugin_path() -> PathBuf {
13 PathBuf::from(env!("RYFT_PJRT_PLUGIN_CUDA_12"))
14}
15
16#[cfg(feature = "cuda-13")]
17pub fn pjrt_cuda_13_plugin_path() -> PathBuf {
18 PathBuf::from(env!("RYFT_PJRT_PLUGIN_CUDA_13"))
19}
20
21#[cfg(feature = "rocm-7")]
22pub fn pjrt_rocm_7_plugin_path() -> PathBuf {
23 PathBuf::from(env!("RYFT_PJRT_PLUGIN_ROCM_7"))
24}
25
26#[cfg(feature = "tpu")]
27pub fn pjrt_tpu_plugin_path() -> PathBuf {
28 PathBuf::from(env!("RYFT_PJRT_PLUGIN_TPU"))
29}
30
31#[cfg(feature = "neuron")]
32pub fn pjrt_neuron_plugin_path() -> PathBuf {
33 PathBuf::from(env!("RYFT_PJRT_PLUGIN_NEURON"))
34}
35
36#[cfg(feature = "metal")]
37pub fn pjrt_metal_plugin_path() -> PathBuf {
38 PathBuf::from(env!("RYFT_PJRT_PLUGIN_METAL"))
39}
40
41pub mod bindings;
42pub mod distributed;
43pub mod protos;