kaio-core 0.2.1

KAIO core — PTX IR types and emission. Part of the KAIO GPU kernel authoring framework.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

//! PTX instruction category enums.
//!
//! Each category is a separate module with an enum that gets populated
//! in its corresponding sprint:
//! - [`ArithOp`] — Sprint 1.2
//! - [`MemoryOp`] — Sprint 1.3
//! - [`ControlOp`] — Sprint 1.4
//!
//! Sprint 1.4 will also add a `special` module with helper functions
//! for thread/block/grid index register access.

pub mod arith;
pub mod control;
pub mod memory;
pub mod special;
pub mod tensor_core;

pub use arith::{ArithOp, MadMode};
pub use control::{CmpOp, ControlOp};
pub use memory::MemoryOp;
pub use tensor_core::{MmaShape, TensorCoreOp};