Revision f309b82a20358a6a6560b87f8325f73ab06c3123 authored by Jevin Jiang on 05 August 2024, 18:06:42 UTC, committed by jax authors on 06 August 2024, 06:17:11 UTC
we also emulate shuffled store using (store + shuffled load + store) for previous generations. PiperOrigin-RevId: 659612665
1 parent f255fb7
| File | Mode | Size |
|---|---|---|
| filecheck | ||
| mosaic | ||
| notebooks | ||
| pallas | ||
| testdata | ||
| third_party | ||
| BUILD | -rw-r--r-- | 30.7 KB |
| ann_test.py | -rw-r--r-- | 7.1 KB |
| aot_test.py | -rw-r--r-- | 3.7 KB |
| api_test.py | -rw-r--r-- | 344.7 KB |
| api_util_test.py | -rw-r--r-- | 2.4 KB |
| array_api_skips.txt | -rw-r--r-- | 1.9 KB |
| array_api_test.py | -rw-r--r-- | 7.8 KB |
| array_interoperability_test.py | -rw-r--r-- | 13.2 KB |
| array_test.py | -rw-r--r-- | 55.0 KB |
| attrs_test.py | -rw-r--r-- | 18.6 KB |
| batching_test.py | -rw-r--r-- | 51.4 KB |
| blocked_sampler_test.py | -rw-r--r-- | 3.3 KB |
| cache_key_test.py | -rw-r--r-- | 11.3 KB |
| checkify_test.py | -rw-r--r-- | 40.6 KB |
| cholesky_update_test.py | -rw-r--r-- | 2.4 KB |
| clear_backends_test.py | -rw-r--r-- | 1.2 KB |
| compilation_cache_test.py | -rw-r--r-- | 20.8 KB |
| config_test.py | -rw-r--r-- | 3.0 KB |
| core_test.py | -rw-r--r-- | 22.8 KB |
| custom_linear_solve_test.py | -rw-r--r-- | 15.9 KB |
| custom_object_test.py | -rw-r--r-- | 10.0 KB |
| custom_root_test.py | -rw-r--r-- | 8.2 KB |
| debug_nans_test.py | -rw-r--r-- | 7.1 KB |
| debugger_test.py | -rw-r--r-- | 11.0 KB |
| debugging_primitives_test.py | -rw-r--r-- | 36.5 KB |
| deprecation_test.py | -rw-r--r-- | 2.0 KB |
| device_test.py | -rw-r--r-- | 1.8 KB |
| dtypes_test.py | -rw-r--r-- | 37.0 KB |
| dynamic_api_test.py | -rw-r--r-- | 60.7 KB |
| errors_test.py | -rw-r--r-- | 11.8 KB |
| experimental_rnn_test.py | -rw-r--r-- | 6.5 KB |
| export_back_compat_test.py | -rw-r--r-- | 34.5 KB |
| export_harnesses_multi_platform_test.py | -rw-r--r-- | 7.3 KB |
| export_test.py | -rw-r--r-- | 71.7 KB |
| extend_test.py | -rw-r--r-- | 6.5 KB |
| fft_test.py | -rw-r--r-- | 16.7 KB |
| for_loop_test.py | -rw-r--r-- | 13.6 KB |
| fused_attention_stablehlo_test.py | -rw-r--r-- | 17.2 KB |
| generated_fun_test.py | -rw-r--r-- | 8.0 KB |
| gpu_memory_flags_test.py | -rw-r--r-- | 1.7 KB |
| heap_profiler_test.py | -rw-r--r-- | 1.2 KB |
| host_callback_test.py | -rw-r--r-- | 108.6 KB |
| host_callback_to_tf_test.py | -rw-r--r-- | 8.6 KB |
| image_test.py | -rw-r--r-- | 14.5 KB |
| infeed_test.py | -rw-r--r-- | 4.1 KB |
| jax_jit_test.py | -rw-r--r-- | 8.0 KB |
| jax_to_ir_test.py | -rw-r--r-- | 5.5 KB |
| jaxpr_effects_test.py | -rw-r--r-- | 31.4 KB |
| jaxpr_util_test.py | -rw-r--r-- | 3.7 KB |
| jet_test.py | -rw-r--r-- | 16.6 KB |
| key_reuse_test.py | -rw-r--r-- | 23.5 KB |
| lax_autodiff_test.py | -rw-r--r-- | 47.1 KB |
| lax_control_flow_test.py | -rw-r--r-- | 98.5 KB |
| lax_metal_test.py | -rw-r--r-- | 225.2 KB |
| lax_numpy_einsum_test.py | -rw-r--r-- | 12.1 KB |
| lax_numpy_indexing_test.py | -rw-r--r-- | 63.5 KB |
| lax_numpy_operators_test.py | -rw-r--r-- | 34.5 KB |
| lax_numpy_reducers_test.py | -rw-r--r-- | 36.2 KB |
| lax_numpy_test.py | -rw-r--r-- | 243.4 KB |
| lax_numpy_ufuncs_test.py | -rw-r--r-- | 12.5 KB |
| lax_numpy_vectorize_test.py | -rw-r--r-- | 9.2 KB |
| lax_scipy_sparse_test.py | -rw-r--r-- | 15.9 KB |
| lax_scipy_special_functions_test.py | -rw-r--r-- | 10.1 KB |
| lax_scipy_spectral_dac_test.py | -rw-r--r-- | 2.0 KB |
| lax_scipy_test.py | -rw-r--r-- | 22.5 KB |
| lax_test.py | -rw-r--r-- | 157.0 KB |
| lax_vmap_op_test.py | -rw-r--r-- | 2.9 KB |
| lax_vmap_test.py | -rw-r--r-- | 29.5 KB |
| layout_test.py | -rw-r--r-- | 17.8 KB |
| lazy_loader_test.py | -rw-r--r-- | 1.3 KB |
| linalg_test.py | -rw-r--r-- | 80.5 KB |
| lobpcg_test.py | -rw-r--r-- | 14.6 KB |
| logging_test.py | -rw-r--r-- | 4.7 KB |
| lru_cache_test.py | -rw-r--r-- | 4.3 KB |
| memories_test.py | -rw-r--r-- | 52.2 KB |
| mesh_utils_test.py | -rw-r--r-- | 29.3 KB |
| metadata_test.py | -rw-r--r-- | 4.1 KB |
| mock_gpu_test.py | -rw-r--r-- | 2.0 KB |
| monitoring_test.py | -rw-r--r-- | 6.2 KB |
| mosaic_test.py | -rw-r--r-- | 1.1 KB |
| multi_device_test.py | -rw-r--r-- | 12.3 KB |
| multibackend_test.py | -rw-r--r-- | 7.3 KB |
| multiprocess_gpu_test.py | -rw-r--r-- | 21.1 KB |
| mutable_array_test.py | -rw-r--r-- | 6.4 KB |
| name_stack_test.py | -rw-r--r-- | 21.3 KB |
| nn_test.py | -rw-r--r-- | 21.8 KB |
| ode_test.py | -rw-r--r-- | 8.5 KB |
| optimizers_test.py | -rw-r--r-- | 10.0 KB |
| package_structure_test.py | -rw-r--r-- | 1.9 KB |
| pgle_test.py | -rw-r--r-- | 9.3 KB |
| pickle_test.py | -rw-r--r-- | 6.1 KB |
| pjit_test.py | -rw-r--r-- | 173.3 KB |
| pmap_test.py | -rw-r--r-- | 117.5 KB |
| polynomial_test.py | -rw-r--r-- | 4.3 KB |
| pretty_printer_test.py | -rw-r--r-- | 1.2 KB |
| profiler_test.py | -rw-r--r-- | 9.7 KB |
| python_callback_test.py | -rw-r--r-- | 36.9 KB |
| pytorch_interoperability_test.py | -rw-r--r-- | 5.6 KB |
| qdwh_test.py | -rw-r--r-- | 7.3 KB |
| random_lax_test.py | -rw-r--r-- | 56.3 KB |
| random_test.py | -rw-r--r-- | 58.5 KB |
| scipy_fft_test.py | -rw-r--r-- | 4.3 KB |
| scipy_interpolate_test.py | -rw-r--r-- | 2.3 KB |
| scipy_ndimage_test.py | -rw-r--r-- | 5.3 KB |
| scipy_optimize_test.py | -rw-r--r-- | 6.8 KB |
| scipy_signal_test.py | -rw-r--r-- | 16.2 KB |
| scipy_spatial_test.py | -rw-r--r-- | 13.8 KB |
| scipy_stats_test.py | -rw-r--r-- | 60.5 KB |
| shape_poly_test.py | -rw-r--r-- | 152.9 KB |
| shard_alike_test.py | -rw-r--r-- | 8.1 KB |
| shard_map_test.py | -rw-r--r-- | 80.8 KB |
| source_info_test.py | -rw-r--r-- | 1.7 KB |
| sourcemap_test.py | -rw-r--r-- | 2.3 KB |
| sparse_bcoo_bcsr_test.py | -rw-r--r-- | 77.0 KB |
| sparse_nm_test.py | -rw-r--r-- | 7.0 KB |
| sparse_test.py | -rw-r--r-- | 47.5 KB |
| sparsify_test.py | -rw-r--r-- | 23.0 KB |
| stack_test.py | -rw-r--r-- | 1.5 KB |
| state_test.py | -rw-r--r-- | 66.0 KB |
| stax_test.py | -rw-r--r-- | 8.0 KB |
| svd_test.py | -rw-r--r-- | 10.8 KB |
| transfer_guard_test.py | -rw-r--r-- | 8.5 KB |
| tree_util_test.py | -rw-r--r-- | 42.6 KB |
| typing_test.py | -rw-r--r-- | 5.8 KB |
| util_test.py | -rw-r--r-- | 6.2 KB |
| version_test.py | -rw-r--r-- | 6.6 KB |
| x64_context_test.py | -rw-r--r-- | 4.4 KB |
| xla_bridge_test.py | -rw-r--r-- | 14.0 KB |
| xla_interpreter_test.py | -rw-r--r-- | 1.1 KB |
Computing file changes ...
