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);