mirrors/ZLUDA
0
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-09-27 03:39:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Remove debugging hack

Browse source
This commit is contained in:
Andrzej Janik 2025-09-18 17:39:31 +00:00
commit 28b464b816
2 changed files with 1 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

30
ptx/src/pass/filter_for_demo.rs
View file

@ -1,30 +0,0 @@
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
}

5
ptx/src/pass/mod.rs
View file

@ -13,7 +13,6 @@ use strum_macros::EnumIter;
mod deparamize_functions; mod deparamize_functions;
mod expand_operands; mod expand_operands;
mod filter_for_demo;
mod fix_special_registers; mod fix_special_registers;
mod hoist_globals; mod hoist_globals;
mod insert_explicit_load_store; mod insert_explicit_load_store;
@ -66,9 +65,7 @@ pub fn to_llvm_module<'input>(
let mut flat_resolver = GlobalStringIdentResolver2::<'input>::new(SpirvWord(1)); let mut flat_resolver = GlobalStringIdentResolver2::<'input>::new(SpirvWord(1));
let mut scoped_resolver = ScopedResolver::new(&mut flat_resolver); let mut scoped_resolver = ScopedResolver::new(&mut flat_resolver);
let sreg_map = SpecialRegistersMap::new(&mut scoped_resolver)?; let sreg_map = SpecialRegistersMap::new(&mut scoped_resolver)?;
let directives = filter_for_demo::run(ast.directives); let directives = normalize_identifiers2::run(&mut scoped_resolver, ast.directives)?;
on_pass_end("filter_for_demo");
let directives = normalize_identifiers2::run(&mut scoped_resolver, directives)?;
on_pass_end("normalize_identifiers2"); on_pass_end("normalize_identifiers2");
let directives = replace_known_functions::run(&mut flat_resolver, directives); let directives = replace_known_functions::run(&mut flat_resolver, directives);
on_pass_end("replace_known_functions"); on_pass_end("replace_known_functions");