Skip to content

GitLab

Commit bcf3a398 authored by rusty1s's avatar rusty1s
Browse files

windows fix

parent 25ef71d2
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 0 deletions
csrc/scatter.cpp
View file @ bcf3a398
#include <Python.h>
#include <torch/script.h> #include <torch/script.h>
#include "cpu/scatter_cpu.h" #include "cpu/scatter_cpu.h"
...@@ -7,6 +8,14 @@ ...@@ -7,6 +8,14 @@
#include "cuda/scatter_cuda.h" #include "cuda/scatter_cuda.h"
#endif #endif
#ifdef _WIN32
#if PY_MAJOR_VERSION < 3
PyMODINIT_FUNC init_C(void) { return NULL; }
#else
PyMODINIT_FUNC PyInit__C(void) { return NULL; }
#endif
#endif
std::tuple<torch::Tensor, torch::optional<torch::Tensor>> std::tuple<torch::Tensor, torch::optional<torch::Tensor>>
scatter_fw(torch::Tensor src, torch::Tensor index, int64_t dim, scatter_fw(torch::Tensor src, torch::Tensor index, int64_t dim,
torch::optional<torch::Tensor> optional_out, torch::optional<torch::Tensor> optional_out,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment