[pallas:mosaic_gpu] Removed leftover debugging code

PiperOrigin-RevId: 703054113
jax-ml · Dec 5, 2024 · 03861d4 · 03861d4
1 parent c965ffb
commit 03861d4
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/jax/_src/pallas/mosaic_gpu/lowering.py b/jax/_src/pallas/mosaic_gpu/lowering.py
@@ -1046,8 +1046,6 @@ def _get_lowering_rule(ctx: LoweringRuleContext, x_smem, *leaves, tree):
   x_smem, transforms = _handle_reshaping(x_smem, transforms)
   x_smem, transforms = _handle_indexing(x_smem, transforms)
 
-  print("ctx:", ctx)
-  print("transforms:", transforms)
   match transforms:
     case (gpu_core.UnswizzleRef(swizzle), gpu_core.UntileRef(tiling)):
       if tiling != (64, swizzle // x_aval.dtype.itemsize):