Refactor implicit conversions, explicit ld/st and global hoistingrepass

1 files changed, 45 insertions, 0 deletions

diff --git a/ptx/src/pass/hoist_globals.rs b/ptx/src/pass/hoist_globals.rs
new file mode 100644
index 0000000..753172a
--- /dev/null
+++ b/ptx/src/pass/hoist_globals.rs
@@ -0,0 +1,45 @@
+use super::*;

+

+pub(super) fn run<'input>(

+    directives: Vec<Directive2<'input, ast::Instruction<SpirvWord>, SpirvWord>>,

+) -> Result<Vec<Directive2<'input, ast::Instruction<SpirvWord>, SpirvWord>>, TranslateError> {

+    let mut result = Vec::with_capacity(directives.len());

+    for mut directive in directives.into_iter() {

+        run_directive(&mut result, &mut directive);

+        result.push(directive);

+    }

+    Ok(result)

+}

+

+fn run_directive<'input>(

+    result: &mut Vec<Directive2<'input, ptx_parser::Instruction<SpirvWord>, SpirvWord>>,

+    directive: &mut Directive2<'input, ptx_parser::Instruction<SpirvWord>, SpirvWord>,

+) -> Result<(), TranslateError> {

+    match directive {

+        Directive2::Variable(..) => {}

+        Directive2::Method(function2) => run_function(result, function2),

+    }

+    Ok(())

+}

+

+fn run_function<'input>(

+    result: &mut Vec<Directive2<'input, ptx_parser::Instruction<SpirvWord>, SpirvWord>>,

+    function: &mut Function2<'input, ptx_parser::Instruction<SpirvWord>, SpirvWord>,

+) {

+    function.body = function.body.take().map(|statements| {

+        statements

+            .into_iter()

+            .filter_map(|statement| match statement {

+                Statement::Variable(var @ ast::Variable {

+                    state_space:

+                        ast::StateSpace::Global | ast::StateSpace::Const | ast::StateSpace::Shared,

+                    ..

+                }) => {

+                    result.push(Directive2::Variable(ast::LinkingDirective::NONE, var));

+                    None

+                }

+                s => Some(s),

+            })

+            .collect()

+    });

+}