From d311003e8eec47a121ab13b326fc322da57e2b89 Mon Sep 17 00:00:00 2001 From: Joe Fioti Date: Wed, 9 Oct 2024 06:47:19 -0400 Subject: [PATCH] Update elementwise_fusion.rs --- crates/luminal_metal/src/elementwise_fusion.rs | 6 ------ 1 file changed, 6 deletions(-) diff --git a/crates/luminal_metal/src/elementwise_fusion.rs b/crates/luminal_metal/src/elementwise_fusion.rs index 06eae08d..bb8a316d 100644 --- a/crates/luminal_metal/src/elementwise_fusion.rs +++ b/crates/luminal_metal/src/elementwise_fusion.rs @@ -446,7 +446,6 @@ impl FusedElementwiseOp { .collect() }) .collect(); - println!("Stacked: {:?}", stacked_shapes); // Stack index expressions let stacked_index_expressions_partial = stacked_shapes .iter() @@ -481,14 +480,12 @@ impl FusedElementwiseOp { .zip(&stacked_shapes) .map(|(partial, sh)| sh[0].valid_expression().substitute('z', partial).simplify()) .collect::>(); - println!("M"); // Replace in subexpressions let n_subexpressions = subexpressions.len(); for (i, ((subexp, _), stacked_shapes)) in subexpressions.iter_mut().zip(subexp_views).enumerate() { - println!("{i}"); // Index for (i, (ind_exp, val_exp)) in stacked_index_expressions .iter() @@ -501,13 +498,10 @@ impl FusedElementwiseOp { input_regexes.insert(i, Regex::new(&format!(r"input{i}([^0-9]|$)")).unwrap()); input_regexes.get(&i).unwrap() }; - // println!("ORIG: {ind_exp}"); let (ind, val) = ( ind_exp.simplify_cache(simplification_cache), val_exp.simplify_cache(simplification_cache), ); - // println!("\n\n\n\n\n\nSIMP: {ind}"); - println!("S {} -> {}", ind_exp.len(), ind.len()); *subexp = re .replace_all( subexp,