{"record":{"id":"eb54659d2a095521","repo":"jax-ml/jax","slug":"tiling-without-swizzle-is-not-supported","errorCode":null,"errorMessage":"Tiling without swizzle is not supported.","messagePattern":"Tiling without swizzle is not supported\\.","errorType":"exception","errorClass":"NotImplementedError","httpStatus":null,"severity":"error","filePath":"jax/_src/pallas/mosaic_gpu/lowering.py","lineNumber":2150,"sourceCode":"    transform_avals = [transform_avals[0], *transform_avals[2:]]\n  else:\n    swizzle = None\n\n  # We verify tiling against swizzling in memory before transposes flip it.\n  # Note: we do not check len(tiling) == 2 here because subsequent transforms\n  # (e.g. ReshapeTransform) may commute with it to produce 2D tiling.\n  tiling_transform = None\n  if transforms and isinstance(transforms[0], gpu_core.UntilingTransform):\n    tiling_transform = transforms[0]\n  elif (\n      len(transforms) > 1\n      and isinstance(transforms[0], gpu_core.ExtractAliasedRef)\n      and isinstance(transforms[1], gpu_core.UntilingTransform)\n  ):\n    tiling_transform = transforms[1]\n  if tiling_transform is not None:\n    if swizzle is None:\n      raise NotImplementedError(\"Tiling without swizzle is not supported.\")\n    bw = dtypes.itemsize_bits(ctx.avals_out[0].dtype)\n    expected_minor_tiling = swizzle * 8 // bw\n    if tiling_transform.tiling[-1] != expected_minor_tiling:\n      raise NotImplementedError(\n          \"Minor tiling dimension does not fit swizzle: \"\n          f\" expected {expected_minor_tiling}, got {tiling_transform.tiling[-1]}\"\n      )\n\n  x_smem, _, transforms = _handle_transforms(\n      ctx, ctx.avals_in[0], x_ref, transform_avals, transforms,\n      allow_peer_refs=True\n  )\n  del x_ref  # Don't use x_ref anymore. Use x_smem instead!\n\n  is_signed = mgpu_utils.is_signed(dtype)\n\n  if not ctx.avals_out[0].shape:  # The scalar case is simple.\n    val = memref_dialect.load(x_smem, [])","sourceCodeStart":2132,"sourceCodeEnd":2168,"githubUrl":"https://github.com/jax-ml/jax/blob/1e1c6a8fc06dfcd1247076ec5cae4640cea5d7bb/jax/_src/pallas/mosaic_gpu/lowering.py#L2132-L2168","documentation":"Error \"Tiling without swizzle is not supported.\" thrown in jax-ml/jax.","triggerScenarios":"Thrown at jax/_src/pallas/mosaic_gpu/lowering.py:2150 when the library encounters an invalid state.","commonSituations":"See trigger scenarios.","solutions":[],"exampleFix":null,"handlingStrategy":null,"validationCode":null,"typeGuard":null,"tryCatchPattern":null,"preventionTips":[],"tags":[],"backgroundTag":null,"analyzedSha":"1e1c6a8fc06dfcd1247076ec5cae4640cea5d7bb","analyzedAt":"2026-08-27T09:53:25.647Z","schemaVersion":2},"datasetVersion":"2026-08-27T13:17:12.746Z"}