ptx-90-parser 0.4.4

Parse NVIDIA PTX 9.0 assembly into a structured AST and explore modules via a CLI.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

//! Original PTX specification:
//!
//! prefetch{.space}.level                    [a];   // prefetch to data cache
//! prefetch.global.level::eviction_priority  [a];   // prefetch to data cache
//! prefetchu.L1  [a];             // prefetch to uniform cache
//! prefetch{.tensormap_space}.tensormap [a];  // prefetch the tensormap
//! .space =                    { .global, .local };
//! .level =                    { .L1, .L2 };
//! .level::eviction_priority = { .L2::evict_last, .L2::evict_normal };
//! .tensormap_space =          { .const, .param };

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

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

    use serde::Serialize;

    #[derive(Debug, Clone, PartialEq, Serialize)]
    pub enum Space {
        Global, // .global
        Local,  // .local
    }

    #[derive(Debug, Clone, PartialEq, Serialize)]
    pub enum Level {
        L1, // .L1
        L2, // .L2
    }

    #[derive(Debug, Clone, PartialEq, Serialize)]
    pub enum LevelEvictionPriority {
        L2EvictNormal, // .L2::evict_normal
        L2EvictLast,   // .L2::evict_last
    }

    #[derive(Debug, Clone, PartialEq, Serialize)]
    pub enum TensormapSpace {
        Const, // .const
        Param, // .param
    }

    #[derive(Debug, Clone, PartialEq, Spanned, Serialize)]
    pub struct PrefetchSpaceLevel {
        pub space: Option<Space>, // {.space}
        pub level: Level,         // .level
        pub a: AddressOperand,    // [a]
        pub span: Span,
    }

    #[derive(Debug, Clone, PartialEq, Spanned, Serialize)]
    pub struct PrefetchGlobalLevelEvictionPriority {
        pub global: (),                                     // .global
        pub level_eviction_priority: LevelEvictionPriority, // .level::eviction_priority
        pub a: AddressOperand,                              // [a]
        pub span: Span,
    }

    #[derive(Debug, Clone, PartialEq, Spanned, Serialize)]
    pub struct PrefetchuL1 {
        pub l1: (),            // .L1
        pub a: AddressOperand, // [a]
        pub span: Span,
    }

    #[derive(Debug, Clone, PartialEq, Spanned, Serialize)]
    pub struct PrefetchTensormapSpaceTensormap {
        pub tensormap_space: Option<TensormapSpace>, // {.tensormap_space}
        pub tensormap: (),                           // .tensormap
        pub a: AddressOperand,                       // [a]
        pub span: Span,
    }
}

// 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::Level as Level0;
pub use section_0::LevelEvictionPriority as LevelEvictionPriority0;
pub use section_0::PrefetchGlobalLevelEvictionPriority;
pub use section_0::PrefetchSpaceLevel;
pub use section_0::PrefetchTensormapSpaceTensormap;
pub use section_0::PrefetchuL1;
pub use section_0::Space as Space0;
pub use section_0::TensormapSpace as TensormapSpace0;