diff --git a/cpp/bench/ann/CMakeLists.txt b/cpp/bench/ann/CMakeLists.txt index ee84f7515a..f29d32ccde 100644 --- a/cpp/bench/ann/CMakeLists.txt +++ b/cpp/bench/ann/CMakeLists.txt @@ -266,7 +266,10 @@ if(RAFT_ANN_BENCH_USE_RAFT_CAGRA) RAFT_CAGRA PATH bench/ann/src/raft/raft_benchmark.cu - $<$:bench/ann/src/raft/raft_cagra.cu> + $<$:bench/ann/src/raft/raft_cagra_float.cu> + $<$:bench/ann/src/raft/raft_cagra_half.cu> + $<$:bench/ann/src/raft/raft_cagra_int8_t.cu> + $<$:bench/ann/src/raft/raft_cagra_uint8_t.cu> LINKS raft::compiled ) diff --git a/cpp/bench/ann/src/raft/raft_cagra_float.cu b/cpp/bench/ann/src/raft/raft_cagra_float.cu new file mode 100644 index 0000000000..058f5bf34a --- /dev/null +++ b/cpp/bench/ann/src/raft/raft_cagra_float.cu @@ -0,0 +1,20 @@ +/* + * Copyright (c) 2023-2024, NVIDIA CORPORATION. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +#include "raft_cagra_wrapper.h" + +namespace raft::bench::ann { +template class RaftCagra; +} // namespace raft::bench::ann diff --git a/cpp/bench/ann/src/raft/raft_cagra_half.cu b/cpp/bench/ann/src/raft/raft_cagra_half.cu new file mode 100644 index 0000000000..a015819ec5 --- /dev/null +++ b/cpp/bench/ann/src/raft/raft_cagra_half.cu @@ -0,0 +1,20 @@ +/* + * Copyright (c) 2023-2024, NVIDIA CORPORATION. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +#include "raft_cagra_wrapper.h" + +namespace raft::bench::ann { +template class RaftCagra; +} // namespace raft::bench::ann diff --git a/cpp/bench/ann/src/raft/raft_cagra_int8_t.cu b/cpp/bench/ann/src/raft/raft_cagra_int8_t.cu new file mode 100644 index 0000000000..be3b83ee60 --- /dev/null +++ b/cpp/bench/ann/src/raft/raft_cagra_int8_t.cu @@ -0,0 +1,20 @@ +/* + * Copyright (c) 2023-2024, NVIDIA CORPORATION. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +#include "raft_cagra_wrapper.h" + +namespace raft::bench::ann { +template class RaftCagra; +} // namespace raft::bench::ann diff --git a/cpp/bench/ann/src/raft/raft_cagra.cu b/cpp/bench/ann/src/raft/raft_cagra_uint8_t.cu similarity index 85% rename from cpp/bench/ann/src/raft/raft_cagra.cu rename to cpp/bench/ann/src/raft/raft_cagra_uint8_t.cu index c0c1352a43..c9679e404d 100644 --- a/cpp/bench/ann/src/raft/raft_cagra.cu +++ b/cpp/bench/ann/src/raft/raft_cagra_uint8_t.cu @@ -17,7 +17,4 @@ namespace raft::bench::ann { template class RaftCagra; -template class RaftCagra; -template class RaftCagra; -template class RaftCagra; } // namespace raft::bench::ann