jax-ml/jax
Documented errors, page 19 of 23. Back to jax-ml/jax
| Code / Message | Type | Severity | Tags |
|---|---|---|---|
| Semaphore value shape | exception | error | pallas, semaphore, shape-mismatch, state-discharge, jax |
| sign_lowering_helper not implemented for | exception | error | jax, pallas, triton, dtype, sign |
| must be positive. | validation | error | jax, pallas, tpu, sliding-window, config-validation |
| Stored array has layout | exception | error | mosaic, gpu, tcgen05, layout, type-check, tensor-memory, jax |
| subclass can't set both `jvp = jvp_from_lin` and `lin… | validation | error | jax, autodiff, circular-definition, misconfiguration |
| Swap does not support masked scalar stores | exception | error | jax, pallas, tpu, sparsecore, atomic, swap |
| Unexpected mask shape | validation | error | jax, splash-attention, mask, shape-validation |
| Unsupported load reduce operation | validation | error | jax, pallas, tmem, reduction, invalid-argument |
| where argument must have dtype=bool; got dtype= | validation | error | jax, ufunc, reduce, dtype, where-mask |
| You can't use two different VectorSubcoreMeshes. | validation | error | jax, pallas, tpu, sparsecore, mesh-compat |
| out of range for indices with | exception | error | jax, sparse, bcoo, vmap, batching |
| c_api argument to load_pjrt_plugin is not a pjrt_c_api… | validation | error | jax, pjrt, plugin, capsule, validation |
| data batch dimensions not compatible for | exception | error | jax, sparse, bcoo, shape-validation, constructor |
| Found an unbound axis name | exception | error | jax, mesh, shard-map, unbound-axis |
| grid_names must have the same length as grid, got | validation | error | jax, pallas, mosaic-gpu, launch-config, validation, shape-mismatch |
| in_specs containing reduced | exception | error | shard-map, reduced, sharding-mismatch, jax |
| jnp.pad: constant_values has unsupported shape | exception | error | jnp-pad, constant-values, invalid-argument-shape |
| does not accept integer axis_name. Got axis_name= | exception | error | jax, axis-name, collectives, api-misuse |
| Number of processes must be defined. | validation | error | jax, distributed, world-size, config |
| Offset is not divisible by tile size | exception | error | jax, mosaic, gpu, tiling, alignment |
| Primitive requires varying manual axes to match, but got … | validation | error | jax, shard-map, spmd, internal-error, workaround |
| reduce only supported for binary ufuncs | validation | error | jax, ufunc, reduce, api-misuse |
| tcgen05_commit_arrive only allows arriving on a Barrier… | exception | error | pallas, mosaic-gpu, tcgen05, barrier, tensor-core |
| The number of query heads must be a multiple of key/value… | validation | error | jax, nn, attention, gqa, shape-validation |
| The process id of the current process must be defined. | validation | error | jax, distributed, rank, config |
| Unimplemented transforms for TMEM refs. | validation | error | jax, pallas, tmem, transforms, not-implemented |
| B tiling too small. Increase swizzle or transpose the input. | validation | error | gpu, mosaic, tcgen05, sparsity, swizzle, not-implemented |
| expected A and E to be the same shape, got A.shape= | validation | error | jax, linalg, frechet-derivative, shape-mismatch |
| expected rank of group_sizes to be >=1. | validation | error | jax, ragged-dot-general, group-sizes, shape-mismatch |
| Indices in subset_by_index must be non-negative. | exception | error | jax, tpu, svd, linalg, negative-index |
| Invalid value ( ) encountered in sharded computation. | exception | error | shard-map, nan, inf, debug-flags, jax |
| jnp.interp: complex x values not supported. | exception | error | jax, interp, complex-dtype, unsupported-operation |
| jnp.pad: with has unsupported shape . Valid shapes are ( … | exception | error | jnp-pad, invalid-argument-shape |
| does not accept dtype . Support for narrow-width integers… | exception | error | jax, lax, narrow-int, int4, dtype-validation |
| Only 1D vectors are supported | validation | error | mosaic-gpu, vector-slice, rank, not-implemented |
| Only unstack of size 2 is supported in Triton. | exception | error | jax, triton, pallas, unstack, not-implemented |
| precv currently only supports manual sharding | exception | error | jax, precv, shard-map, manual-sharding |
| shape is not a valid result of applying tiling . | validation | error | mosaic, fragmented-array, tiling, shape-mismatch |
| Stored array has dtype | exception | error | mosaic, gpu, tcgen05, dtype-mismatch, tensor-memory, jax |
| Transpose with multiref is not supported. | exception | error | jax, transpose, multiref |
| Unsupported element type for block scaling | validation | error | gpu, mosaic, tcgen05, block-scaling, not-implemented |
| Unsupported TPU device kind | exception | error | tpu, cpu-testing, abstract-mesh, device-kind, jax |
| Axis must be specified when shapes of a and weights differ. | validation | error | jax, average, weights, shape-mismatch |
| Bitcast ref with dynamic size is not supported. | exception | error | jax, bitcast, dynamic-shape |
| Expected 0D shape, but got | validation | error | jax, pallas, scalar, shape-mismatch |
| Explicit sharding inference for ragged_dot_general is not… | exception | error | jax, ragged-dot, sharding, multi-device, spmd |
| Got unexpected `to` value. Allowed `to` values are | exception | error | jax, pcast, invalid-argument |
| jnp.insert(): obj must be a slice, a one-dimensional array… | exception | error | jax, numpy-compat, insert, ndim |
| Masks must have the same shape | validation | error | jax, splash-attention, mask, shape-mismatch |
| Must wait an int32 value. | exception | error | pallas, semaphore, dtype-validation, jax |
| __pow__ modulo not implemented | exception | error | jax, shape-polymorphism, pow, not-implemented |
| reduction operation does not have an identity, so to use a… | validation | error | jax, reductions, where-mask, identity |
| Requested too many CUDA threads per block. Each Mosaic… | validation | error | jax, pallas, mosaic-gpu, thread-limits, resource-limit, validation |
| Tiling does not apply to shape | validation | error | mosaic, fragmented-array, tiling, shape-mismatch |
| TracerBoolConversionError | exception | error | jax, tracer, bool-conversion, control-flow, jit |
| a and b must have the same element type, but got | exception | error | jax, pallas, triton, dtype, matmul |
| All arrays must have the same dtype, got | validation | error | jax, mosaic-gpu, validation, dtype-mismatch |
| All elements in the grid must be strictly positive, but got | exception | error | jax, pallas, grid, validation |
| should not be greater than | validation | error | jax, tpu, pallas, splash-attention, block-sizes, validation |
| c_api argument to clear_xla_transform_c_api is not a… | validation | error | jax, xla, pjrt, pycapsule |
| Can only concatenate 1D vectors | validation | error | mosaic-gpu, vector-concat, rank |
| dot_general requires equal numbers of lhs_batch and… | exception | error | jax, dot-general, batch-dims, dimension-numbers |
| Expected a ShapeDtypeStruct, but got | validation | error | jax, pallas, shapedtypestruct, return-type |
| Failed to infer a possible set of layouts. This should only… | exception | error | jax, mosaic, layout, unsatisfiable-constraints |
| Got empty index range in subset_by_index. | exception | error | jax, tpu, svd, linalg, off-by-one |
| `index_type` must be i32. | exception | error | jax, pallas, triton, argmax, index-dtype |
| Indexing barrier with | exception | error | jax, mosaic-gpu, barrier, transforms |
| inline_mgpu_p only supports plgpu.ShapeDtypeStruct return… | validation | error | jax, pallas, mosaic-gpu, inline-mgpu, type-validation |
| lax.platform_dependent: the | validation | error | jax, lax, platform-dependent, typeerror, callable |
| Mesh axis names cannot be None. Got | validation | error | jax, mesh, axis-names, input-validation |
| noverlap must be less than nperseg. | exception | error | jax, scipy, stft, overlap, argument-validation |
| Only 1 reduction dimension is supported. | exception | error | jax, mosaic, gpu, reduction |
| overwrite_data argument not implemented. | exception | error | jax, scipy, detrend, not-implemented, immutable-arrays |
| reduce_axes argument to transpose is deprecated | validation | error | jax, linear-transpose, deprecated-api |
| reduce_window got inconsistent window_strides and… | validation | error | jax, shape-validation, reduce-window, strides |
| SymbolicScope constraint must be a string: got | exception | error | jax, shape-polymorphism, constraints, type-validation |
| Unexpected precision specifier value | exception | error | jax, lstm, rnn, invalid-argument, precision |
| Unsupported dtype for reduction | exception | error | mosaic, gpu, tcgen05, dtype, reduction, jax |
| Unsupported mesh type | exception | error | pallas, mosaic-gpu, mesh, interpret-mode, type-validation |
| Unsupported TPU chip version | exception | error | tpu, chip-version, unsupported-hardware, jax |
| Unsupported transform | exception | error | jax, mosaic, transform, not-implemented |
| batch size must be even when megacore_mode is 'batch' | validation | error | jax, pallas, tpu, paged-attention, megacore, batch-size |
| cluster_names must have the same length as cluster, got | validation | error | jax, pallas, mosaic-gpu, launch-config, validation, shape-mismatch |
| Could not deserialize PyTreeDefProto. | exception | error | jax, pytree, protobuf, corrupt-data |
| dynamic_slice: indices must be scalars or slices. Got index… | validation | error | jax, indexing, boolean-index, dynamic-slice |
| load_staged_rhs must be an integer or None. | exception | error | jax, pallas, tpu, matmul, type-error |
| Logic for `nan_policy` of | exception | error | jax, scipy, stats, mode, not-implemented, nan-policy |
| Nesting MultiHeadMasks is not supported | validation | error | jax, splash-attention, mask, multi-head, nesting |
| philox4x32_random_bits got invalid prng key. | exception | error | jax, prng, philox, key-validation |
| Please install the `filelock` package to set… | error_code | error | jax, compilation-cache, optional-dependency, filelock |
| ragged_dot_general requires that group_sizes.dtype is… | validation | error | jax, ragged-dot, dtype-validation |
| Stride is not divisible by (tile size = ). Strides: , tiling | exception | error | jax, mosaic, gpu, tiling, strides, alignment |
| transpose of bcoo_extract with assume_unique=False | exception | error | jax, sparse, bcoo, autodiff, transpose |
| `unreduced_kind` should be `None` when `unreduced` is an… | validation | error | jax, sharding, partition-spec, validation |
| unsupported atomic operation | exception | error | jax, pallas, triton, atomics |
| A, X must have same dtypes | exception | error | jax, sparse, lobpcg, dtype |
| Allocated too many names | exception | error | profiling, mosaic, gpu, resource-exhaustion |
| B scale shape mismatch: expected | validation | error | gpu, mosaic, tcgen05, shape-mismatch, block-scaling |
| Cannot split a Pallas key. Use fold_in instead to generate… | exception | error | jax, pallas, tpu, random, prng, not-implemented |
| Index is out of bounds for packing factor | exception | error | jax, pallas, tpu, index-out-of-bounds |