first tutorial

Author: kwea123

.vscode/c_cpp_properties.json

@@ -0,0 +1,19 @@
+{
+    "configurations": [
+        {
+            "name": "Linux",
+            "includePath": [
+                "${workspaceFolder}/**",
+                "/home/ubuntu/anaconda3/envs/cppcuda/include/python3.8",
+                "/home/ubuntu/anaconda3/envs/cppcuda/lib/python3.8/site-packages/torch/include",
+                "/home/ubuntu/anaconda3/envs/cppcuda/lib/python3.8/site-packages/torch/include/torch/csrc/api/include"
+            ],
+            "defines": [],
+            "compilerPath": "/usr/bin/clang",
+            "cStandard": "c17",
+            "cppStandard": "c++14",
+            "intelliSenseMode": "linux-clang-x64"
+        }
+    ],
+    "version": 4
+}

interpolation.cpp

@@ -0,0 +1,14 @@
+#include <torch/extension.h>
+
+
+torch::Tensor trilinear_interpolation(
+    torch::Tensor feats,
+    torch::Tensor point
+){
+    return feats;
+}
+
+
+PYBIND11_MODULE(TORCH_EXTENSION_NAME, m){
+    m.def("trilinear_interpolation", &trilinear_interpolation);
+}

setup.py

@@ -0,0 +1,20 @@
+from setuptools import setup
+from torch.utils.cpp_extension import CppExtension, BuildExtension
+
+
+setup(
+    name='cppcuda_tutorial',
+    version='1.0',
+    author='kwea123',
+    author_email='kwea123@gmail.com',
+    description='cppcuda example',
+    long_description='cppcuda example',
+    ext_modules=[
+        CppExtension(
+            name='cppcuda_tutorial',
+            sources=['interpolation.cpp'])
+    ],
+    cmdclass={
+        'build_ext': BuildExtension
+    }
+)

test.py

@@ -0,0 +1,11 @@
+import torch
+import cppcuda_tutorial
+
+
+feats = torch.ones(2)
+point = torch.zeros(2)
+
+
+out = cppcuda_tutorial.trilinear_interpolation(feats, point)
+
+print(out)