rocm
Functions
| Function | Description |
|---|---|
rocm_has_gpu | |
rocm_add | |
rocm_mm | |
rocm_scale | |
rocm_dot | |
rocm_relu |
Details
rocm_has_gpu
fn rocm_has_gpu() -> i64rocm_add
fn rocm_add(a: i64, b: i64, c: i64, len: i64) -> i64rocm_mm
fn rocm_mm(a: i64, b: i64, c: i64, M: i64, N: i64, K: i64) -> i64rocm_scale
fn rocm_scale(a: i64, c: i64, scalar: i64, len: i64) -> i64rocm_dot
fn rocm_dot(a: i64, b: i64, len: i64) -> i64rocm_relu
fn rocm_relu(a: i64, c: i64, len: i64) -> i64