https://github.com/google/jax
History
Tip revision: 46368e4e734938bffe7978cb4077923343095abc authored by Tianjian Lu on 04 November 2022, 04:39:10 UTC
[sparse] Update the guard of cusparse SpMM and SpMv algorithms to cusparse version 11.7.1 onwards.
Tip revision: 46368e4
File Mode Size
debugger.gif -rw-r--r-- 424.2 KB
device_memory_profile.svg -rw-r--r-- 19.7 KB
device_memory_profile_leak1.svg -rw-r--r-- 14.2 KB
device_memory_profile_leak2.svg -rw-r--r-- 12.4 KB
favicon.png -rw-r--r-- 6.5 KB
jax_logo_250px.png -rw-r--r-- 33.2 KB
mesh.jpg -rw-r--r-- 48.0 KB
multi_host.jpg -rw-r--r-- 38.1 KB
partition_spec_none_y.png -rw-r--r-- 23.8 KB
partition_spec_x_none.png -rw-r--r-- 13.3 KB
partition_spec_x_y.png -rw-r--r-- 6.8 KB
partition_spec_xy.png -rw-r--r-- 7.3 KB
partition_spec_y_none.png -rw-r--r-- 21.8 KB
perfetto.png -rw-r--r-- 96.5 KB
style.css -rw-r--r-- 77 bytes
tensorboard_profiler.png -rw-r--r-- 29.1 KB
type_lattice.svg -rw-r--r-- 22.3 KB
vscode-completion.png -rw-r--r-- 30.8 KB
xla_spmd.jpg -rw-r--r-- 41.3 KB

back to top