ptx
Functions
| Function | Description |
|---|---|
ptx_has_gpu | |
ptx_vecadd | |
ptx_matmul | |
ptx_scale | |
ptx_dot | |
ptx_relu |
Details
ptx_has_gpu
fn ptx_has_gpu() -> i64ptx_vecadd
fn ptx_vecadd(a: i64, b: i64, c: i64, len: i64) -> i64ptx_matmul
fn ptx_matmul(a: i64, b: i64, c: i64, M: i64, N: i64, K: i64) -> i64ptx_scale
fn ptx_scale(a: i64, c: i64, scalar: i64, len: i64) -> i64ptx_dot
fn ptx_dot(a: i64, b: i64, len: i64) -> i64ptx_relu
fn ptx_relu(a: i64, c: i64, len: i64) -> i64