Qiao Zhang 5d7f639769 Add small and big matmul to api_benchmarks. ...

name                                  cpu/op
jit_small_matmul                      2.96µs ± 2%
jit_big_matmul                        22.1µs ±21%

name                                  time/op

jit_small_matmul                      2.96µs ± 2%
jit_big_matmul                        22.7µs ±21%

PiperOrigin-RevId: 435453853

2022-03-17 14:51:17 -07:00

..

api_benchmark.py

Add small and big matmul to api_benchmarks.

2022-03-17 14:51:17 -07:00

benchmark.py

[JAX] Fix pylint errors.

2021-10-04 17:54:46 -07:00

gda_benchmark.py

Add the cache back now that Mesh's __hash__ is also being hashed on self.devices.shape.

2022-02-01 14:06:01 -08:00

pmap_benchmark.py

Revert import path changes to examples/ and benchmarks/

2021-01-20 17:35:55 -08:00