Skip to main content

cuda_capabilities

Function cuda_capabilities 

Source
pub const fn cuda_capabilities() -> &'static [OperationCapability]
Expand description

Return the CUDA backend operation capability descriptor table.

ยงExamples

use tenferro_core_ops::PrimitiveOpKind;
use tenferro_gpu::cuda::cuda_capabilities;
use tenferro_tensor::{DType, SupportLevel};

let add = cuda_capabilities()
    .iter()
    .find(|entry| entry.op == PrimitiveOpKind::Add && entry.dtype == DType::I32)
    .unwrap();
assert_eq!(add.result, SupportLevel::Native);