jax-ml/jax
Documented errors, page 14 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| inline_mgpu in a single-warp context only supports scalar… | validation | error | jax, pallas, inline-mgpu, warp-semantics, scalar-required |
| Only 2-argument stack is supported in Triton. | exception | error | jax, triton, pallas, stack, not-implemented |
| Only the eigvals=None case of eigh is implemented. | validation | error | jax, linalg, eigenvalue, not-implemented |
| Transfer is not a multiple of | exception | error | jax, pallas, mosaic-gpu, alignment, warpgroup, async-store |
| tree mismatch during linearization of | error_code | error | jax, pytree, linearize, autodiff |
| x and y cannot be broadcast together. | exception | error | jax, scipy, csd, broadcasting, shape-mismatch |
| All input tensors must have the same rank. Got lhs rank | exception | error | jax, scaled-dot, rank-validation |
| Barrier does not support arbitrary transforms | validation | error | mosaic-gpu, pallas, barrier, transforms |
| does not accept dtype at position . Accepted dtypes at… | exception | error | jax, lax, binop, dtype-validation, type-error |
| Expected 'key' head dimension to be | validation | error | jax, pallas, tpu, splash-attention, shape, head-dim |
| Loads from TMEM layout | exception | error | mosaic, gpu, tcgen05, layout, tensor-memory, jax |
| method argument to `loggamma` must be one of | validation | error | jax, random, loggamma, method, input-validation |
| nfft must be greater than or equal to nperseg. | exception | error | jax, scipy, stft, fft-size, argument-validation |
| No profile started | error_code | error | jax, profiler, lifecycle |
| No VJP defined for custom_vjp function | exception | error | jax, custom-vjp, missing-rule, autodiff |
| `out_sharding` passed to `dot_general` can only contain… | validation | error | jax, dot-general, sharding, spmd, experimental-api |
| ref must be scalar. | validation | error | jax, pallas, triton, atomic-cas, shape |
| shard_map requires a `jax.sharding.Mesh` or a… | exception | error | jax, shard-map, mesh, type-error, distributed |
| Swizzle mismatch. In transforms swizzle | validation | error | jax, mosaic-gpu, transpose, swizzle, transforms |
| transpose permutation cannot permute batch axes with… | exception | error | jax, sparse, bcoo, transpose, permutation, batching |
| unexpected input | validation | error | jax, dtype, validation |
| Unknown core type | exception | error | tpu, pallas, core-type, version-mismatch, mlir |
| All args passed to `explicit_layout` must have the same… | validation | error | jax, layout, consistency |
| bcoo_multiply_sparse: arrays must have same number of… | exception | error | jax, sparse, bcoo, shape-mismatch, elementwise-multiply |
| Dimension must be 0 for 1D iota. | exception | error | jax, pallas, tpu, iota, indexing |
| Duplicate serialization registration for serialized_name | validation | error | jax, export, serialization, registry, name-collision |
| (set JAX_PLATFORMS='' to automatically choose an available… | exception | error | |
| Expected WGMMAAbstractAccumulatorRef got | validation | error | mosaic-gpu, pallas, wgmma, accumulator, type-error |
| Factor is only used in compound factors; must specify its… | exception | error | jax, sharding, compound-factor, factor-sizes |
| fill_value must be a scalar or tuple of scalars; got | validation | error | jax, nonzero, fill-value, argument-validation |
| Input sharding mesh should be equal to out_sharding mesh | validation | error | jax, mesh, all-gather, shard-map |
| Input tiling is not compatible with | validation | error | jax, mosaic-gpu, collapse-shape, tiling, unsatisfiable |
| Integer indexing of refs that follows a non-trivial slice… | exception | error | jax, pallas, tpu, sparsecore, indexing |
| Invalid scalar value | validation | error | jax, scalar, type-check |
| Invalid WGMMA accumulator dtype for s8/i8 WGMMA. Expected… | exception | error | jax, pallas, wgmma, int8, dtype |
| jnp. : where must be None or a boolean array; got . | validation | error | jax, reductions, where-mask, dtype |
| len(a) must be at least 1; got shape | exception | error | jax, convolution-matrix, empty-array |
| multi_dot requires at least two arrays; got len(arrays)= | exception | error | jax, numpy, linalg, matmul, argument-validation |
| reduce_axes argument to vjp is deprecated | validation | error | jax, vjp, deprecated, migration |
| `unroll` must be a `bool` or a non-negative `int`. | validation | error | jax, scan, unroll, valueerror, argument-validation |
| Unsupported shape | validation | error | jax, pallas, shape-validation, ndim |
| Arguments to after_all must be tokens | validation | error | jax, lax, tokens, control-flow, xla |
| batch dimension sizes must match; got | exception | error | jax, shape-mismatch, searchsorted, batching |
| __dlpack__ device only supported for TPU pinned host memory | exception | error | jax, dlpack, tpu, pinned-memory, interop |
| dtype argument to `rayleigh` must be a float dtype, got | validation | error | jax, random, dtype-validation |
| expected non-empty vector for x | validation | error | jax, numpy, polynomial, polyfit, empty-array, edge-case |
| index can't contain negative values | exception | error | jnp-pad, negative-pad-width |
| Invalid axis for num_programs | exception | error | jax, pallas, mosaic, grid, axis, lowering |
| Invalid in/out transforms. In transform | validation | error | jax, mosaic-gpu, transpose, tiling, consistency |
| jax.numpy.block does not allow empty list arguments | validation | error | jax, block, empty-list, valueerror |
| jnp.linalg.cond: input array must be at least 2D; got | exception | error | jax, numpy, linalg, condition-number, shape-validation |
| Loads are only allowed on VMEM and SMEM references. | validation | error | jax, pallas, tpu, memory-space, async-copy |
| Num arrivals must be at least 1, but got | validation | error | jax, pallas, barrier, validation, config |
| out_dtype must be integer typed; got | exception | error | jax, dtype, nonzero, argument-validation |
| `page_indices` and `q` must have the same batch size | validation | error | jax, pallas, tpu, paged-attention, shape-validation |
| Passing an array as a dtype argument is no longer… | validation | error | jax, dtype, api-removal, migration |
| remat optimization for custom_vjp does not support… | exception | error | jax, custom-vjp, remat, higher-order-ad |
| reps length must be equal to the ndim of x, got | validation | error | jax, tile, shape-validation |
| Rule cannot handle an op with vector operands or results | validation | error | mosaic, gpu, lowering, layout-mismatch, not-implemented |
| Shape dimension must be divisible by | validation | error | jax, pallas, tpu, random, philox, alignment |
| shard_map requires a callable for its first argument, but… | exception | error | jax, shard-map, callable, type-error |
| contracting dim of size must be divisible by its scale's… | exception | error | jax, scaled-dot, shape-validation, quantization |
| Strided slices not implemented | validation | error | jax, mosaic-gpu, memref, slice, not-implemented, strided |
| structure of output returned by custom vmap rule | validation | error | jax, custom-vmap, pytree, vmap |
| Warp reduction group size should be a power of 2 | validation | error | mosaic-gpu, warp-reduction, power-of-two |
| args argument to jax.scipy.optimize.minimize must be a… | exception | error | jax, scipy, optimize, type-error, argument-validation |
| Base indices of array slices must be aligned to the… | validation | error | mosaic, gpu, slicing, tile-alignment |
| Can only load a single key per load. | validation | error | jax, pallas, tpu, prng, indexing |
| Cannot handle new consts created by state discharge. | exception | error | pallas, state-discharge, run-scoped, not-implemented, jax |
| Cannot make an array with dtype | exception | error | jax, string-dtype, dtype-mismatch |
| cummax: x= must be rank 1 | exception | error | jax, pallas, sparsecore, cummax, rank |
| dimension_semantics must be the same length as grid. | exception | error | jax, pallas, dimension-semantics, validation |
| Dynamic grid bounds not supported in the Mosaic GPU… | exception | error | |
| Expected an index-typed index | validation | error | jax, mosaic-gpu, memref, mlir, type-mismatch |
| Expected WGSplatFragLayout, but got | validation | error | jax, pallas, layout, scalar, mosaic-gpu |
| masked load_p | validation | error | jax, pallas, tpu, interpret-mode, not-implemented, masked-load |
| Mismatch in result shapes. | exception | error | jax, custom-partitioning, sharding, shape-mismatch |
| num_threads and thread_name must be either both set or both… | validation | error | jax, pallas, mosaic-gpu, warp-specialization, validation |
| Offset must be scalar, got | validation | error | jax, pallas, tpu, random, philox, scalar-validation |
| Only support memref.cast where the input and output types… | exception | error | jax, mosaic-gpu, memref, cast, type-mismatch |
| ormqr with left=False expects c to have the same number of… | validation | error | jax, linalg, ormqr, shape-validation, qr |
| PJRT plugin config file should contain "library_path" field. | exception | error | |
| Seed key_data must be 1D. | validation | error | jax, pallas, tpu, prng, shape-validation |
| Unexpected layout for | validation | error | jax, pallas, inline-mgpu, layout-mismatch |
| Unhandled transforms for multimem_load_reduce | exception | error | jax, pallas, transforms, multimem, lowering |
| Uninitialized Refs are not supported in lowering of… | exception | error | jax, pallas, run-state, uninitialized-ref |
| Unsupported dtype for reduction | validation | error | jax, pallas, tmem, dtype, reduction |
| WGMMA accumulators can only be allocated non-collectively… | exception | error | jax, pallas, wgmma, run-scoped, collective-axes |
| Can't compute input and output sizes of a | validation | error | jax, nn, initializers, shape-validation |
| Cannot transpose with respect to sparse indices | exception | error | jax, sparse, csr, autodiff, gradient |
| Columns must be a multiple of 16, got | exception | error | jax, mosaic, tmem, layout, alignment |
| DMA start priority must be non-negative | exception | error | jax, pallas, mosaic, dma, priority, argument-validation |
| dot_general requires contracting dimensions to have the… | validation | error | jax, dot-general, shape-mismatch, contraction |
| Error writing persistent compilation cache entry for | console | warning | jax, compilation-cache, io, permissions |
| expected search dim * 5 < matrix dim | exception | error | jax, sparse, lobpcg, validation |
| Finalize is not implemented | exception | error | jax, collectives, cpu, api-misuse |
| initial value must be a scalar. Got array of shape | validation | error | jax, reductions, initial-value, shape-mismatch |
| Mosaic kernels cannot be automatically partitioned. Please… | exception | error | tpu, pallas, sharding, shard-map, jax |
| Swizzle= is too big for MMA with M=64. Try lowering it. | validation | error | jax, mosaic, gpu, mma, swizzle, tile-size, tcgen05 |
| tcgen05_mma only allows arriving on a Barrier | exception | error | pallas, mosaic-gpu, tcgen05, mma, barrier, interpret-mode |