fix poor performance of slogdet when input dimension n=1 or 2#710

cj401-amd wants to merge 2 commits intorocm-jaxlib-v0.8.2ROCm/jax:rocm-jaxlib-v0.8.2from

ci_cj_slogdet_small_rocm-jaxlib-v0.8.2ROCm/jax:ci_cj_slogdet_small_rocm-jaxlib-v0.8.2Copy head branch name to clipboard