update

pyg-team · Jan 6, 2025 · dd3c040 · dd3c040
1 parent 50cd40f
commit dd3c040
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/pyg_lib/csrc/classes/cpu/hash_map.cpp b/pyg_lib/csrc/classes/cpu/hash_map.cpp
@@ -0,0 +1,15 @@
+#include <torch/library.h>
+
+namespace pyg {
+namespace classes {
+
+struct CPUHashMap : torch::CustomClassHolder {
+  CPUHashMap(){};
+};
+
+TORCH_LIBRARY(pyg, m) {
+  m.class_<CPUHashMap>("CPUHashMap").def(torch::init());
+}
+
+}  // namespace classes
+}  // namespace pyg
diff --git a/test/csrc/classes/test_hash_map.cpp b/test/csrc/classes/test_hash_map.cpp
@@ -0,0 +1,7 @@
+#include "pyg_lib/csrc/classes/cpu/hash_map.cpp"
+
+#include <gtest/gtest.h>
+
+TEST(CPUHashMapTest, BasicAssertions) {
+  auto map = pyg::classes::CPUHashMap();
+}