optimize_softmax

Optimize softmax in triton in many cases

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
outer_softmax.py		outer_softmax.py
outer_softmax_online.py		outer_softmax_online.py
softmax_loop_along_reduce_axis_v1.py		softmax_loop_along_reduce_axis_v1.py
softmax_loop_along_reduce_axis_v2.py		softmax_loop_along_reduce_axis_v2.py
softmax_naive.py		softmax_naive.py
softmax_online_v1.py		softmax_online_v1.py
softmax_online_v2.py		softmax_online_v2.py
softmax_online_v2_evict.py		softmax_online_v2_evict.py
softmax_online_v2_rev.py		softmax_online_v2_rev.py
softmax_online_v2_spec.py		softmax_online_v2_spec.py
softmax_online_v2_spec_rev.py		softmax_online_v2_spec_rev.py
softmax_online_v2_spec_rev_evict.py		softmax_online_v2_spec_rev_evict.py
softmax_split.py		softmax_split.py

Provide feedback