From 03861d43ec4063dd82850b6cdac4438e3d5d4e94 Mon Sep 17 00:00:00 2001 From: Sergei Lebedev Date: Thu, 5 Dec 2024 03:38:24 -0800 Subject: [PATCH] [pallas:mosaic_gpu] Removed leftover debugging code PiperOrigin-RevId: 703054113 --- jax/_src/pallas/mosaic_gpu/lowering.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/jax/_src/pallas/mosaic_gpu/lowering.py b/jax/_src/pallas/mosaic_gpu/lowering.py index 2900da133158..79f8f116fbd4 100644 --- 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):