ptx-90-parser 0.3.0

Parse NVIDIA PTX 9.0 assembly into a structured AST and explore modules via a CLI.
Documentation
//! Original PTX specification:
//!
//! // 32-bit scalar operation, with optional secondary operation
//! vop.dtype.atype.u32{.sat}.mode       d, a{.asel}, b{.bsel};
//! vop.dtype.atype.u32{.sat}.mode.op2   d, a{.asel}, b{.bsel}, c;
//! // 32-bit scalar operation, with optional data merge
//! vop.dtype.atype.u32{.sat}.mode  d.dsel, a{.asel}, b{.bsel}, c;
//! vop   = { vshl, vshr };
//! .dtype = .atype = { .u32, .s32 };
//! .mode  = { .clamp, .wrap };
//! .dsel  = .asel  = .bsel  = { .b0, .b1, .b2, .b3, .h0, .h1 };
//! .op2   = { .add, .min, .max };

#![allow(unused)]
use crate::r#type::common::*;

pub mod section_0 {
    use crate::r#type::common::*;

    #[derive(Debug, Clone, PartialEq)]
    pub enum Dtype {
        U32, // .u32
        S32, // .s32
    }

    #[derive(Debug, Clone, PartialEq)]
    pub enum Atype {
        U32, // .u32
        S32, // .s32
    }

    #[derive(Debug, Clone, PartialEq)]
    pub enum Mode {
        Clamp, // .clamp
        Wrap,  // .wrap
    }

    #[derive(Debug, Clone, PartialEq)]
    pub enum Asel {
        B0, // .b0
        B1, // .b1
        B2, // .b2
        B3, // .b3
        H0, // .h0
        H1, // .h1
    }

    #[derive(Debug, Clone, PartialEq)]
    pub enum Bsel {
        B0, // .b0
        B1, // .b1
        B2, // .b2
        B3, // .b3
        H0, // .h0
        H1, // .h1
    }

    #[derive(Debug, Clone, PartialEq)]
    pub enum Op2 {
        Add, // .add
        Min, // .min
        Max, // .max
    }

    #[derive(Debug, Clone, PartialEq)]
    pub enum Dsel {
        B0, // .b0
        B1, // .b1
        B2, // .b2
        B3, // .b3
        H0, // .h0
        H1, // .h1
    }

    #[derive(Debug, Clone, PartialEq)]
    pub struct VshlDtypeAtypeU32SatMode {
        pub dtype: Dtype,       // .dtype
        pub atype: Atype,       // .atype
        pub u32: (),            // .u32
        pub sat: bool,          // {.sat}
        pub mode: Mode,         // .mode
        pub d: GeneralOperand,  // d
        pub a: GeneralOperand,  // a
        pub asel: Option<Asel>, // {.asel}
        pub b: GeneralOperand,  // b
        pub bsel: Option<Bsel>, // {.bsel}
    }

    #[derive(Debug, Clone, PartialEq)]
    pub struct VshrDtypeAtypeU32SatMode {
        pub dtype: Dtype,       // .dtype
        pub atype: Atype,       // .atype
        pub u32: (),            // .u32
        pub sat: bool,          // {.sat}
        pub mode: Mode,         // .mode
        pub d: GeneralOperand,  // d
        pub a: GeneralOperand,  // a
        pub asel: Option<Asel>, // {.asel}
        pub b: GeneralOperand,  // b
        pub bsel: Option<Bsel>, // {.bsel}
    }

    #[derive(Debug, Clone, PartialEq)]
    pub struct VshlDtypeAtypeU32SatModeOp2 {
        pub dtype: Dtype,       // .dtype
        pub atype: Atype,       // .atype
        pub u32: (),            // .u32
        pub sat: bool,          // {.sat}
        pub mode: Mode,         // .mode
        pub op2: Op2,           // .op2
        pub d: GeneralOperand,  // d
        pub a: GeneralOperand,  // a
        pub asel: Option<Asel>, // {.asel}
        pub b: GeneralOperand,  // b
        pub bsel: Option<Bsel>, // {.bsel}
        pub c: GeneralOperand,  // c
    }

    #[derive(Debug, Clone, PartialEq)]
    pub struct VshrDtypeAtypeU32SatModeOp2 {
        pub dtype: Dtype,       // .dtype
        pub atype: Atype,       // .atype
        pub u32: (),            // .u32
        pub sat: bool,          // {.sat}
        pub mode: Mode,         // .mode
        pub op2: Op2,           // .op2
        pub d: GeneralOperand,  // d
        pub a: GeneralOperand,  // a
        pub asel: Option<Asel>, // {.asel}
        pub b: GeneralOperand,  // b
        pub bsel: Option<Bsel>, // {.bsel}
        pub c: GeneralOperand,  // c
    }

    #[derive(Debug, Clone, PartialEq)]
    pub struct VshlDtypeAtypeU32SatMode1 {
        pub dtype: Dtype,       // .dtype
        pub atype: Atype,       // .atype
        pub u32: (),            // .u32
        pub sat: bool,          // {.sat}
        pub mode: Mode,         // .mode
        pub d: GeneralOperand,  // d
        pub dsel: Dsel,         // .dsel
        pub a: GeneralOperand,  // a
        pub asel: Option<Asel>, // {.asel}
        pub b: GeneralOperand,  // b
        pub bsel: Option<Bsel>, // {.bsel}
        pub c: GeneralOperand,  // c
    }

    #[derive(Debug, Clone, PartialEq)]
    pub struct VshrDtypeAtypeU32SatMode1 {
        pub dtype: Dtype,       // .dtype
        pub atype: Atype,       // .atype
        pub u32: (),            // .u32
        pub sat: bool,          // {.sat}
        pub mode: Mode,         // .mode
        pub d: GeneralOperand,  // d
        pub dsel: Dsel,         // .dsel
        pub a: GeneralOperand,  // a
        pub asel: Option<Asel>, // {.asel}
        pub b: GeneralOperand,  // b
        pub bsel: Option<Bsel>, // {.bsel}
        pub c: GeneralOperand,  // c
    }
}

// Re-export types with section suffixes to avoid naming conflicts
// e.g., Type0 for section_0::Type, Type1 for section_1::Type
pub use section_0::Asel as Asel0;
pub use section_0::Atype as Atype0;
pub use section_0::Bsel as Bsel0;
pub use section_0::Dsel as Dsel0;
pub use section_0::Dtype as Dtype0;
pub use section_0::Mode as Mode0;
pub use section_0::Op2 as Op20;
pub use section_0::VshlDtypeAtypeU32SatMode;
pub use section_0::VshlDtypeAtypeU32SatMode1;
pub use section_0::VshlDtypeAtypeU32SatModeOp2;
pub use section_0::VshrDtypeAtypeU32SatMode;
pub use section_0::VshrDtypeAtypeU32SatMode1;
pub use section_0::VshrDtypeAtypeU32SatModeOp2;