-
Notifications
You must be signed in to change notification settings - Fork 113
Open
Description
The following code seems to account for only a subset of the valid conditions:
kokkos-kernels/blas/tpls/KokkosBlas1_dot_tpl_spec_decl.hpp
Lines 132 to 140 in 9bca19c
| #define KOKKOSBLAS1_DOT_TPL_SPEC_DECL_CUBLAS_EXT(ETI_SPEC_AVAIL) \ | |
| KOKKOSBLAS1_DOT_TPL_SPEC_DECL_CUBLAS(Kokkos::LayoutLeft, float, float, Kokkos::Cuda, Kokkos::CudaSpace, cublasSdot, \ | |
| ETI_SPEC_AVAIL) \ | |
| KOKKOSBLAS1_DOT_TPL_SPEC_DECL_CUBLAS(Kokkos::LayoutLeft, double, double, Kokkos::Cuda, Kokkos::CudaSpace, \ | |
| cublasDdot, ETI_SPEC_AVAIL) \ | |
| KOKKOSBLAS1_DOT_TPL_SPEC_DECL_CUBLAS(Kokkos::LayoutLeft, Kokkos::complex<float>, cuComplex, Kokkos::Cuda, \ | |
| Kokkos::CudaSpace, cublasCdotc, ETI_SPEC_AVAIL) \ | |
| KOKKOSBLAS1_DOT_TPL_SPEC_DECL_CUBLAS(Kokkos::LayoutLeft, Kokkos::complex<double>, cuDoubleComplex, Kokkos::Cuda, \ | |
| Kokkos::CudaSpace, cublasZdotc, ETI_SPEC_AVAIL) |
Indeed, both
Kokkos::View<double*, Kokkos::Cuda, Kokkos::LayoutLeft>and
Kokkos::View<double*, Kokkos::Cuda, Kokkos::LayoutRight>should map to cublasDdot since the underlying memory is the same.
Stated in other words, the condition should probably read
Is the view layout with unit stride ?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels