From fcf3cd70229cdc729d05ddab081ac886c9db6bd7 Mon Sep 17 00:00:00 2001 From: Tom Rix Date: Fri, 29 Sep 2023 13:58:28 -0700 Subject: [PATCH] torch python 3.12 changes Signed-off-by: Tom Rix --- functorch/csrc/dim/dim.cpp | 6 ++++++ torch/csrc/dynamo/cpython_defs.h | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/functorch/csrc/dim/dim.cpp b/functorch/csrc/dim/dim.cpp index b611dc3e8c2..c7009478aee 100644 --- a/functorch/csrc/dim/dim.cpp +++ b/functorch/csrc/dim/dim.cpp @@ -10,7 +10,13 @@ // Many APIs have changed/don't exist anymore #if IS_PYTHON_3_12_PLUS +#include "dim.h" + // Re-enable this some day +PyObject* Dim_init() { + PyErr_SetString(PyExc_RuntimeError, "First class dim doesn't work with python 3.12"); + return nullptr; +} #else diff --git a/torch/csrc/dynamo/cpython_defs.h b/torch/csrc/dynamo/cpython_defs.h index f0a0e1a88e2..f58becd246e 100644 --- a/torch/csrc/dynamo/cpython_defs.h +++ b/torch/csrc/dynamo/cpython_defs.h @@ -6,7 +6,7 @@ // should go in cpython_defs.c. Copying is required when, e.g., // we need to call internal CPython functions that are not exposed. -#if IS_PYTHON_3_11_PLUS && !(IS_PYTHON_3_12_PLUS) +#if IS_PYTHON_3_11_PLUS #include -- 2.43.0