support torch 2.3 (#94)

* support torch 2.3

* bump version to 1.0.1

src/sfast/__init__.py

@@ -32,4 +32,4 @@ def new_lru_cache(*args, **kwargs):
 
 # This line will be programatically read/write by setup.py.
 # Leave them at the bottom of this file and don't touch them.
-__version__ = "1.0.0"
+__version__ = "1.0.1"

src/sfast/csrc/operators/cublas/cublas_gemm.cpp

@@ -38,6 +38,12 @@
 #include "CUDABlas.h"
 #include "cublas_gemm.h"
 
+#if TORCH_VERSION_MAJOR >= 2 && TORCH_VERSION_MINOR >= 3
+namespace c10 {
+static auto in_place = std::in_place;
+}
+#endif
+
 namespace sfast {
 namespace operators {
 

src/sfast/csrc/operators/fused_linear.cpp

@@ -10,8 +10,15 @@
 #include <c10/util/MaybeOwned.h>
 #include <c10/util/irange.h>
 #include <torch/library.h>
+
 #include "fused_linear.h"
 
+#if TORCH_VERSION_MAJOR >= 2 && TORCH_VERSION_MINOR >= 3
+namespace c10 {
+static auto in_place = std::in_place;
+}
+#endif
+
 namespace sfast {
 namespace operators {
 

version.txt

@@ -1 +1 @@
-1.0.0
+1.0.1