import `torch/torch.h` instead of `torch/extension.h` (#260)

27633e2e · Gerico Vidanes · GitHub · ad228bb8 · 27633e2e · 27633e2e
Unverified Commit 27633e2e authored Jul 23, 2022 by Gerico Vidanes Committed by GitHub Jul 23, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

csrc/extensions.h csrc/extensions.h +1 -1

csrc/sparse.h csrc/sparse.h +1 -2

No files found.
--- a/csrc/extensions.h
+++ b/csrc/extensions.h
 #include "macros.h"
-#include <torch/extension.h>
+#include <torch/torch.h>
--- a/csrc/sparse.h
+++ b/csrc/sparse.h
 #pragma once

-#include <torch/extension.h>
-
+#include "extensions.h"
 #include "macros.h"

 namespace sparse {