windows fix

bcf3a398 · rusty1s · 25ef71d2 · bcf3a398
Commit bcf3a398 authored Feb 11, 2020 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

csrc/scatter.cpp csrc/scatter.cpp +9 -0

No files found.
--- a/csrc/scatter.cpp
+++ b/csrc/scatter.cpp
+#include <Python.h>
 #include <torch/script.h>

 #include "cpu/scatter_cpu.h"
@@ -7,6 +8,14 @@
 #include "cuda/scatter_cuda.h"
 #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>>
 scatter_fw(torch::Tensor src, torch::Tensor index, int64_t dim,
           torch::optional<torch::Tensor> optional_out,