ptx_parser/unparser/instruction/
cp_reduce_async_bulk_tensor.rs

1//! Original PTX specification:
2//!
3//! // shared::cta -> global
4//! cp.reduce.async.bulk.tensor.dim.dst.src.redOp{.load_mode}.completion_mechanism{.level::cache_hint} [tensorMap, tensorCoords], [srcMem] {,cache-policy};
5//! .dst =                  { .global };
6//! .src =                  { .shared::cta };
7//! .dim =                  { .1d, .2d, .3d, .4d, .5d };
8//! .completion_mechanism = { .bulk_group };
9//! .load_mode =            { .tile, .im2col_no_offs };
10//! .redOp =                { .add, .min, .max, .inc, .dec, .and, .or, .xor};
11
12#![allow(unused)]
13
14use crate::lexer::PtxToken;
15use crate::unparser::{PtxUnparser, common::*};
16
17pub mod section_0 {
18    use super::*;
19    use crate::r#type::instruction::cp_reduce_async_bulk_tensor::section_0::*;
20
21    impl PtxUnparser for CpReduceAsyncBulkTensorDimDstSrcRedopLoadModeCompletionMechanismLevelCacheHint {
22        fn unparse_tokens(&self, tokens: &mut ::std::vec::Vec<PtxToken>) {
23            push_opcode(tokens, "cp");
24                    push_directive(tokens, "reduce");
25                    push_directive(tokens, "async");
26                    push_directive(tokens, "bulk");
27                    push_directive(tokens, "tensor");
28                    match &self.dim {
29                            Dim::_1d => {
30                                    push_directive(tokens, "1d");
31                            }
32                            Dim::_2d => {
33                                    push_directive(tokens, "2d");
34                            }
35                            Dim::_3d => {
36                                    push_directive(tokens, "3d");
37                            }
38                            Dim::_4d => {
39                                    push_directive(tokens, "4d");
40                            }
41                            Dim::_5d => {
42                                    push_directive(tokens, "5d");
43                            }
44                    }
45                    match &self.dst {
46                            Dst::Global => {
47                                    push_directive(tokens, "global");
48                            }
49                    }
50                    match &self.src {
51                            Src::SharedCta => {
52                                    push_directive(tokens, "shared::cta");
53                            }
54                    }
55                    match &self.redop {
56                            Redop::Add => {
57                                    push_directive(tokens, "add");
58                            }
59                            Redop::Min => {
60                                    push_directive(tokens, "min");
61                            }
62                            Redop::Max => {
63                                    push_directive(tokens, "max");
64                            }
65                            Redop::Inc => {
66                                    push_directive(tokens, "inc");
67                            }
68                            Redop::Dec => {
69                                    push_directive(tokens, "dec");
70                            }
71                            Redop::And => {
72                                    push_directive(tokens, "and");
73                            }
74                            Redop::Xor => {
75                                    push_directive(tokens, "xor");
76                            }
77                            Redop::Or => {
78                                    push_directive(tokens, "or");
79                            }
80                    }
81                    if let Some(load_mode_0) = self.load_mode.as_ref() {
82                            match load_mode_0 {
83                                    LoadMode::Im2colNoOffs => {
84                                            push_directive(tokens, "im2col_no_offs");
85                                    }
86                                    LoadMode::Tile => {
87                                            push_directive(tokens, "tile");
88                                    }
89                            }
90                    }
91                    match &self.completion_mechanism {
92                            CompletionMechanism::BulkGroup => {
93                                    push_directive(tokens, "bulk_group");
94                            }
95                    }
96                    if self.level_cache_hint {
97                            push_directive(tokens, "level::cache_hint");
98                    }
99                    self.tensormap.unparse_tokens(tokens);
100            tokens.push(PtxToken::Comma);
101                    self.srcmem.unparse_tokens(tokens);
102            if self.cache_policy.is_some() { tokens.push(PtxToken::Comma); }
103                    if let Some(opt_1) = self.cache_policy.as_ref() {
104                        opt_1.unparse_tokens(tokens);
105                    }
106            tokens.push(PtxToken::Semicolon);
107        }
108    }
109
110}
111
ptx_parser/unparser/instruction/cp_reduce_async_bulk_tensor.rs

ptx_parser/unparser/instruction/
cp_reduce_async_bulk_tensor.rs
