mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/cpu
History
Eugene Zhulenev 1e03917c43 [xla:ffi] Use lazy decoding for Buffer<dtype,rank> 
name                old cpu/op   new cpu/op   delta
BM_AnyBufferArgX1   11.0ns ± 3%  11.2ns ±10%   +1.76%  (p=0.000 n=67+69)
BM_AnyBufferArgX4   12.4ns ± 3%  12.4ns ± 4%   -0.31%  (p=0.006 n=69+69)
BM_BufferArgX1      12.5ns ± 1%  11.1ns ± 4%  -11.20%  (p=0.000 n=62+76)
BM_BufferArgX4      19.1ns ± 1%  14.4ns ± 4%  -24.84%  (p=0.000 n=64+73)
BM_BufferArgX8      36.0ns ± 5%  20.3ns ± 4%  -43.59%  (p=0.000 n=79+75)
BM_TupleOfI32Attrs  66.4ns ± 1%  66.4ns ± 2%   -0.03%  (p=0.000 n=66+72)

PiperOrigin-RevId: 650691450
2024-07-09 11:07:25 -07:00
..
_lapack
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
BUILD
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
cpu_kernels.cc
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
lapack_kernels_using_lapack.cc
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
lapack_kernels.cc
[xla:ffi] Use lazy decoding for Buffer<dtype,rank>
2024-07-09 11:07:25 -07:00
lapack_kernels.h
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
lapack.cc
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00
lapack.h
Port Singular Value Decomposition to XLA's FFI
2024-07-08 05:19:53 -07:00