Skip to content

Commit b5bd14b

Browse files
committed
add instantiation and testing type and allow half/bfloat16 from CMake
1 parent 41735db commit b5bd14b

5 files changed

Lines changed: 400 additions & 108 deletions

File tree

CMakeLists.txt

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,9 +90,7 @@ if(MSVC OR MINGW)
9090
FORCE
9191
)
9292
endif()
93-
if(GINKGO_ENABLE_HALF AND GINKGO_ENABLE_BFLOAT16)
94-
message(FATAL_ERROR "We do not support half and bfloat16 at the same time.")
95-
endif()
93+
9694
option(
9795
GINKGO_SKIP_DEPENDENCY_UPDATE
9896
"Do not update dependencies each time the project is rebuilt"

common/cuda_hip/matrix/csr_kernels.instantiate.cpp

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
1+
// SPDX-FileCopyrightText: 2017 - 2025 The Ginkgo authors
22
//
33
// SPDX-License-Identifier: BSD-3-Clause
44

@@ -40,6 +40,12 @@ GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT5(GKO_DECLARE_CSR_SPMV_KERNEL,
4040
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT6(GKO_DECLARE_CSR_SPMV_KERNEL,
4141
int32);
4242
// split
43+
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT7(GKO_DECLARE_CSR_SPMV_KERNEL,
44+
int32);
45+
// split
46+
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT8(GKO_DECLARE_CSR_SPMV_KERNEL,
47+
int32);
48+
// split
4349
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT1(GKO_DECLARE_CSR_SPMV_KERNEL,
4450
int64);
4551
// split
@@ -57,6 +63,12 @@ GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT5(GKO_DECLARE_CSR_SPMV_KERNEL,
5763
// split
5864
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT6(GKO_DECLARE_CSR_SPMV_KERNEL,
5965
int64);
66+
// split
67+
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT7(GKO_DECLARE_CSR_SPMV_KERNEL,
68+
int64);
69+
// split
70+
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT8(GKO_DECLARE_CSR_SPMV_KERNEL,
71+
int64);
6072

6173

6274
// split
@@ -78,6 +90,12 @@ GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT5(
7890
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT6(
7991
GKO_DECLARE_CSR_ADVANCED_SPMV_KERNEL, int32);
8092
// split
93+
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT7(
94+
GKO_DECLARE_CSR_ADVANCED_SPMV_KERNEL, int32);
95+
// split
96+
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT8(
97+
GKO_DECLARE_CSR_ADVANCED_SPMV_KERNEL, int32);
98+
// split
8199
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT1(
82100
GKO_DECLARE_CSR_ADVANCED_SPMV_KERNEL, int64);
83101
// split
@@ -95,6 +113,12 @@ GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT5(
95113
// split
96114
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT6(
97115
GKO_DECLARE_CSR_ADVANCED_SPMV_KERNEL, int64);
116+
// split
117+
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT7(
118+
GKO_DECLARE_CSR_ADVANCED_SPMV_KERNEL, int64);
119+
// split
120+
GKO_INSTANTIATE_FOR_EACH_MIXED_VALUE_TYPE_SPLIT8(
121+
GKO_DECLARE_CSR_ADVANCED_SPMV_KERNEL, int64);
98122

99123

100124
// split

0 commit comments

Comments
 (0)