update

pyg-team · Dec 30, 2024 · 24b8f45 · 24b8f45
1 parent 328681b
commit 24b8f45
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -12,6 +12,11 @@ recursive-exclude third_party/parallel-hashmap/tests *
 recursive-exclude third_party/parallel-hashmap/examples *
 recursive-exclude third_party/parallel-hashmap/benchmark *
 recursive-exclude third_party/METIS/graphs *
+recursive-exclude third_party/cccl/benchmarks *
+recursive-exclude third_party/cccl/docs *
+recursive-exclude third_party/cccl/examples *
+recursive-exclude third_party/cuCollections/examples *
+recursive-exclude third_party/cuCollections/tests *
 
 recursive-exclude test *
 recursive-exclude docs *

diff --git a/pyg_lib/csrc/ops/cuda/sampled_kernel.cu b/pyg_lib/csrc/ops/cuda/sampled_kernel.cu
@@ -2,15 +2,13 @@
 #include <ATen/cuda/CUDAContext.h>
 #include <torch/library.h>
 
-#include <cuco/static_map.cuh>
-
 namespace pyg {
 namespace ops {
 
 namespace {
 
 #define THREADS 1024
-#define CDIV(N, M) ((N) + (M) - 1) / (M)
+#define CDIV(N, M) ((N) + (M)-1) / (M)
 
 enum FnType { ADD, SUB, MUL, DIV };
 const std::map<std::string, FnType> to_fn_type = {