add dice to dispatch_00_generate.py

rapidsai · Jul 1, 2024 · 0fa6b67 · 0fa6b67
1 parent de0f45f
commit 0fa6b67
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cpp/src/distance/detail/pairwise_matrix/dispatch_00_generate.py b/cpp/src/distance/detail/pairwise_matrix/dispatch_00_generate.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2023, NVIDIA CORPORATION.
+# 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.
@@ -15,7 +15,7 @@
 # NOTE: this template is not perfectly formatted. Use pre-commit to get
 # everything in shape again.
 header = """/*
- * Copyright (c) 2021-2023, NVIDIA CORPORATION.
+ * Copyright (c) 2021-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.
@@ -95,6 +95,11 @@
         OpT="raft::distance::detail::ops::cosine_distance_op",
         archs = [60, 80],
     ),
+    dict(
+        path_prefix="dice",
+        OpT="raft::distance::detail::ops::dice_distance_op",
+        archs = [60, 80],
+    ),
     dict(
         path_prefix="hamming_unexpanded",
         OpT="raft::distance::detail::ops::hamming_distance_op",