Skip to content

Commit 03861d4

Browse files
superbobryGoogle-ML-Automation
authored andcommitted
[pallas:mosaic_gpu] Removed leftover debugging code
PiperOrigin-RevId: 703054113
1 parent c965ffb commit 03861d4

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

jax/_src/pallas/mosaic_gpu/lowering.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1046,8 +1046,6 @@ def _get_lowering_rule(ctx: LoweringRuleContext, x_smem, *leaves, tree):
10461046
x_smem, transforms = _handle_reshaping(x_smem, transforms)
10471047
x_smem, transforms = _handle_indexing(x_smem, transforms)
10481048

1049-
print("ctx:", ctx)
1050-
print("transforms:", transforms)
10511049
match transforms:
10521050
case (gpu_core.UnswizzleRef(swizzle), gpu_core.UntileRef(tiling)):
10531051
if tiling != (64, swizzle // x_aval.dtype.itemsize):

0 commit comments

Comments
 (0)