task1: implementations

[fnrizzi]

Scope

The objective would be do things like:

Kokkos:View<double**> A;
Kokkos::par_for(100, 
  KOKKOS_LAMBDA(int i)
  {
    auto Aslice = subview(A...);

    KokkosBlas::SerialGemv(... );
    KokkosBlas::SerialGemm(... );
    KokkosBlas::SerialScal(... );

    KokkosBlas::ThreadVectorGemv(... );
    KokkosBlas::ThreadVectorGemm(... );
  });

The key thing is to have these implemented inside the KokkosBlas namespace.

IMPORTANT: TeamVector from SOW actually means ThreadVector or Vector (something that can run under TeamThreadRange) - not to be confused with TeamVectorRange.

Note: "special" implementations, like atomic-updating gemm in BSR SpGEMM are not in scope of this task.

Update any relevant documentation - where exactly ? (e.g. Wiki)

Status

	Serial	Team / TeamVector	ThreadVector
a) GEMV	✅Done #1433	✅Done #1435	⏳ Queued for review #1556
b) GEMM	✔ In review #1519	✔ In review #1519	⏳ Queued for review #1556
c) Scal	✅Done #1448	✅ Done #1448	⏳ Queued for review #1556

[mzuzek]

@lucbv @fnrizzi

PR Status

To activate PR links click or CTRL+click the chart to open it in new tab/window

---

src

---

Legend

src

Merged PRs

• Move GEMV:
  • Move SerialGEMV to KokkosBlas kokkos/kokkos-kernels#1433
  • Move {Team,TeamVector}Gemv to KokkosBlas kokkos/kokkos-kernels#1435
• Move Set and Scale with their unit test:
  • Move {Serial,Team,TeamVector} Set to KokkosBlas kokkos/kokkos-kernels#1454
  • Move Set/Scale unit test to KokkosBlas kokkos/kokkos-kernels#1455
  • Move {Serial,Team,TeamVector}Scale to KokkosBlas kokkos/kokkos-kernels#1448
• Fixes:
  • Fix incorrect function call in KokkosBatched::TeamGEMV unit test kokkos/kokkos-kernels#1444
  • Fix GitHub CI failing on broken develop kokkos/kokkos-kernels#1461
  • Fix MKL build errors kokkos/kokkos-kernels#1478