PyTorch interface

.gitmodules

@@ -17,3 +17,7 @@
 [submodule "third_party/json"]
  path = third_party/json
  url = https://github.com/nlohmann/json
+
+[submodule "third_party/dlpack"]
+ path = third_party/dlpack
+ url = https://github.com/dmlc/dlpack

.vscode/c_cpp_properties.json

@@ -4,6 +4,7 @@
  "name": "Linux",
  "includePath": [
  "${workspaceFolder}/**",
+ "${workspaceFolder}/third_party/mscclpp/include",
  "/usr/local/cuda/include",
  "/opt/rocm/include"
  ],

ark/CMakeLists.txt

@@ -17,6 +17,7 @@ set(COMMON_LIBS ARK::numa ARK::ibverbs pthread rt)
 target_include_directories(ark_obj PUBLIC ${CMAKE_CURRENT_SOURCE_DIR}/include)
 target_include_directories(ark_obj PRIVATE ${CMAKE_CURRENT_SOURCE_DIR})
 target_include_directories(ark_obj SYSTEM PRIVATE
+ ${DLPACK_INCLUDE_DIRS}
  ${JSON_INCLUDE_DIRS}
  ${MSCCLPP_INCLUDE_DIRS}
  ${IBVERBS_INCLUDE_DIRS}

ark/api/context.cpp

@@ -29,4 +29,8 @@ void Context::set(const std::string& key, const std::string& value,
  this->impl_->set(key, value_json, type);
 }
 
+std::string Context::dump() const {
+ return this->impl_->dump().dump();
+}
+
 } // namespace ark