From f71cd95f7130e43783094108e811e1eedea76a8e Mon Sep 17 00:00:00 2001
From: Andrzej Janik <vosen@vosen.pl>
Date: Fri, 19 Sep 2025 18:17:08 +0000
Subject: [PATCH] demo mode

---
 ptx/src/pass/filter_for_demo.rs | 30 ++++++++++++++++++++++++++++++
 ptx/src/pass/mod.rs             |  5 ++++-
 2 files changed, 34 insertions(+), 1 deletion(-)
 create mode 100644 ptx/src/pass/filter_for_demo.rs

diff --git a/ptx/src/pass/filter_for_demo.rs b/ptx/src/pass/filter_for_demo.rs
new file mode 100644
index 0000000..816aab7
--- /dev/null
+++ b/ptx/src/pass/filter_for_demo.rs
@@ -0,0 +1,30 @@
+pub(crate) fn run<'input>(
+    directives: Vec<ptx_parser::Directive<'input, ptx_parser::ParsedOperand<&'input str>>>,
+) -> Vec<ptx_parser::Directive<'input, ptx_parser::ParsedOperand<&'input str>>> {
+    let demo_kernels_path = std::env::var("ZLUDA_DEMO_KERNELS").unwrap();
+    let demo_kernels_file = std::fs::read_to_string(demo_kernels_path).unwrap();
+    let demo_kernels = demo_kernels_file
+        .lines()
+        .map(|line| line.trim())
+        .filter(|line| !line.is_empty())
+        .collect::<std::collections::HashSet<_>>();
+    let result = directives
+        .into_iter()
+        .filter(|directive| match directive {
+            ptx_parser::Directive::Method(_, method) => {
+                !method.func_directive.name.is_kernel()
+                    || demo_kernels.contains(method.func_directive.name())
+            }
+            _ => true,
+        })
+        .collect::<Vec<_>>();
+    for directive in result.iter() {
+        match directive {
+            ptx_parser::Directive::Method(_, method) => {
+                eprintln!("{}", method.func_directive.name());
+            }
+            _ => {}
+        }
+    }
+    result
+}
diff --git a/ptx/src/pass/mod.rs b/ptx/src/pass/mod.rs
index 4f87dc3..e4a22c7 100644
--- a/ptx/src/pass/mod.rs
+++ b/ptx/src/pass/mod.rs
@@ -13,6 +13,7 @@ use strum_macros::EnumIter;
 
 mod deparamize_functions;
 mod expand_operands;
+mod filter_for_demo;
 mod fix_special_registers;
 mod hoist_globals;
 mod insert_explicit_load_store;
@@ -65,7 +66,9 @@ pub fn to_llvm_module<'input>(
     let mut flat_resolver = GlobalStringIdentResolver2::<'input>::new(SpirvWord(1));
     let mut scoped_resolver = ScopedResolver::new(&mut flat_resolver);
     let sreg_map = SpecialRegistersMap::new(&mut scoped_resolver)?;
-    let directives = normalize_identifiers2::run(&mut scoped_resolver, ast.directives)?;
+    let directives = filter_for_demo::run(ast.directives);
+    on_pass_end("filter_for_demo");
+    let directives = normalize_identifiers2::run(&mut scoped_resolver, directives)?;
     on_pass_end("normalize_identifiers2");
     let directives = replace_known_functions::run(&mut flat_resolver, directives);
     on_pass_end("replace_known_functions");