{"record":{"id":"6043dee9107e9c78","repo":"jax-ml/jax","slug":"unsupported-core-type-core-type","errorCode":null,"errorMessage":"Unsupported core type: {core_type}","messagePattern":"Unsupported core type: (.+?)","errorType":"validation","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"jax/_src/pallas/mosaic/core.py","lineNumber":499,"sourceCode":"    | pallas_core.MemorySpace\n    | pallas_core.CoreMemorySpace\n    | jax_core.MemorySpace\n):\n  match memory_space:\n    case None:\n      match core_type:\n        case CoreType.TC:\n          return pallas_core.MemorySpace.ANY\n        case CoreType.SC_SCALAR_SUBCORE | CoreType.SC_VECTOR_SUBCORE:\n          return MemorySpace.HBM\n    case pallas_core.MemorySpace.DEFAULT:\n      match core_type:\n        case CoreType.TC | CoreType.SC_VECTOR_SUBCORE:\n          return MemorySpace.VMEM\n        case CoreType.SC_SCALAR_SUBCORE:\n          return MemorySpace.SMEM\n        case _:\n          raise ValueError(f\"Unsupported core type: {core_type}\")\n    case pallas_core.MemorySpace.ANY | jax_core.MemorySpace.Host:\n      return memory_space\n    case (\n        pallas_core.MemorySpace.ERROR\n        | pallas_core.MemorySpace.INDEX\n        | pallas_core.MemorySpace.KEY\n    ):\n      return MemorySpace.SMEM\n    case pallas_core.CoreMemorySpace():\n      return (\n          memory_space.memory_space\n          if memory_space.mesh.core_type is core_type\n          else memory_space\n      )\n    case acc if isinstance(acc, AccMemorySpace):\n      return acc\n    case MemorySpace():\n      return memory_space","sourceCodeStart":481,"sourceCodeEnd":517,"githubUrl":"https://github.com/jax-ml/jax/blob/1e1c6a8fc06dfcd1247076ec5cae4640cea5d7bb/jax/_src/pallas/mosaic/core.py#L481-L517","documentation":"memory_space_to_tpu_memory_space maps a Pallas memory space to a TPU-specific space based on the core type. Only TensorCore and SparseCore vector subcore map to VMEM and SparseCore scalar subcore maps to SMEM; any other CoreType value falls through to this ValueError.","triggerScenarios":"Lowering a Mosaic kernel (pipelined/unpipelined lowering, block-mapping checks, load lowering) where a memory space carries a CoreType not in {TC, SC_VECTOR_SUBCORE, SC_SCALAR_SUBCORE}, e.g., a newly added or synthetic core type.","commonSituations":"Version skew between JAX and a newer/older Mosaic/PJRT that introduces new CoreType enum members; custom memory-space annotations with unexpected core types.","solutions":["Update JAX (and its bundled Mosaic) to matching versions so all CoreType members are handled","Avoid annotating refs with core types outside TC / SC vector / SC scalar subcores on TPU","File an issue if a legitimate new core type is unhandled in this match statement"],"exampleFix":null,"handlingStrategy":"type-guard","validationCode":"from jax._src.pallas.mosaic.core import CoreType\nassert core_type in (CoreType.TC, CoreType.SC_VECTOR_SUBCORE, CoreType.SC_SCALAR_SUBCORE)","typeGuard":null,"tryCatchPattern":null,"preventionTips":["Pin JAX and libtpu/Mosaic to compatible versions","Avoid annotating refs with novel core types"],"tags":["jax","pallas","tpu","memory-space","lowering","version-skew"],"backgroundTag":"unsupported-enum-value","analyzedSha":"1e1c6a8fc06dfcd1247076ec5cae4640cea5d7bb","analyzedAt":"2026-08-27T09:53:25.647Z","schemaVersion":2},"datasetVersion":"2026-08-27T13:17:12.746Z"}