From 24b8f45df01d9bd875d3aba8545df1dc303e94ab Mon Sep 17 00:00:00 2001
From: rusty1s <matthias.fey@tu-dortmund.de>
Date: Mon, 30 Dec 2024 20:08:54 +0000
Subject: [PATCH] update

---
 MANIFEST.in                             | 5 +++++
 pyg_lib/csrc/ops/cuda/sampled_kernel.cu | 4 +---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/MANIFEST.in b/MANIFEST.in
index 927383884..450750eae 100644
--- 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
index ae916dc3f..e36a83a32 100644
--- 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 = {