Merge branch 'branch-24.04' into fix_readme

rapidsai · Mar 13, 2024 · 6318d52 · 6318d52
2 parents 4ab8e2c + 9466695
commit 6318d52
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 281 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -31,16 +31,19 @@ repos:
           )
         types_or: [c, c++, cuda]
         args: ["-fallback-style=none", "-style=file", "-i"]
-  - repo: local
-    hooks:
-      - id: copyright-check
-        name: copyright-check
-        entry: python3 ./scripts/checks/copyright.py --update-current-year
-        language: python
-        pass_filenames: false
-        additional_dependencies: [gitpython]
   - repo: https://github.com/rapidsai/dependency-file-generator
     rev: v1.8.0
     hooks:
       - id: rapids-dependency-file-generator
         args: ["--clean"]
+  - repo: https://github.com/rapidsai/pre-commit-hooks
+    rev: v0.0.1
+    hooks:
+      - id: verify-copyright
+        files: |
+          (?x)
+              [.](cmake|cpp|cu|cuh|h|hpp|sh|pxd|py|pyx)$|
+              CMakeLists[.]txt$|
+              meta[.]yaml$|
+              setup[.]cfg$|
+              [.]flake8[.]cython$
diff --git a/ci/checks/copyright.py b/ci/checks/copyright.py
diff --git a/cpp/src/wholememory_ops/functions/exchange_embeddings_nccl_func.cu b/cpp/src/wholememory_ops/functions/exchange_embeddings_nccl_func.cu
@@ -62,8 +62,8 @@ wholememory_error_code_t exchange_embeddings_nccl_func(const void* dev_local_gat
                        WHOLEMEMORY_DT_INT8,
                        stream);
     WM_CUDA_DEBUG_SYNC_STREAM(stream);
-    WHOLEMEMORY_EXPECTS(wm_comm->sync_stream(stream) == WHOLEMEMORY_SUCCESS,
-                        "Embedding AllToAllV failed.");
+    // WHOLEMEMORY_EXPECTS(wm_comm->sync_stream(stream) == WHOLEMEMORY_SUCCESS,
+    //                    "Embedding AllToAllV failed.");
   } catch (wholememory::logic_error& wle) {
     WHOLEMEMORY_ERROR("exchange_embeddings_nccl_func LOGIC Error %s\n", wle.what());
     return WHOLEMEMORY_LOGIC_ERROR;