remove prints

nod-ai · Jan 13, 2025 · 4983ed9 · 4983ed9
1 parent 06827d5
commit 4983ed9
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/sharktank/sharktank/kernels/attention.py b/sharktank/sharktank/kernels/attention.py
@@ -192,8 +192,6 @@ def generate(self, ksel: KernelSelection, kb: KernelBuilder):
             "scale_type": scale_type_str,
             "o_type": o_type_str,
         }
-        print("ARGS:")
-        print(kwargs)
         template_file = "flash_attention.mlir"
         target_function_name = f"sharktank_flash_attention_{l}_{s}_{d}_{e}_{i_type_str}_{scale_type_str}_{o_type_str}"
         target_function = inline_template_function(