support python3.11

python/jittor/__init__.py

@@ -9,7 +9,7 @@
 # file 'LICENSE.txt', which is part of this source code package.
 # ***************************************************************
 
-__version__ = '1.3.7.4'
+__version__ = '1.3.7.5'
 from jittor_utils import lock
 with lock.lock_scope():
     ori_int = int

python/jittor/compiler.py

@@ -1368,7 +1368,7 @@ if has_cuda and is_cuda:
     nvcc_flags = " " + os.environ.get("nvcc_flags", "") + " "
     nvcc_flags += convert_nvcc_flags(cc_flags)
     nvcc_version = list(jit_utils.get_int_version(nvcc_path))
-    max_arch = 1000
+    max_arch = 86
     if nvcc_version < [11,]:
         max_arch = 75
     elif nvcc_version < [11,1]:

python/jittor/src/pybind/py_var_tracer.cc

@@ -35,6 +35,11 @@ static PyObject* my_import(const char* module_name, const char* attr) {
 }
 
 static PyObject* find_obj_name(PyFrameObject* f, PyObject* obj, const char* default_name="_model") {
+    #if PY_MINOR_VERSION>=11
+    #pragma message( "PY_MAJOR_VERSION333 " PY_VERSION )
+    LOGf << "python3.11 not supported yet";
+    return nullptr;
+    #else
     auto co = f->f_code;
     auto map = co->co_varnames;
 
@@ -71,6 +76,7 @@ static PyObject* find_obj_name(PyFrameObject* f, PyObject* obj, const char* defa
     }
     // LOGw << "not found name" << map << co->co_cellvars << co->co_freevars << (PyObject*)f;
     return PyUnicode_FromString(default_name);
+    #endif
 }
 
 static string to_string(PyObject* obj) {
@@ -80,6 +86,10 @@ static string to_string(PyObject* obj) {
 }
 
 static vector<Stack> get_stack_info() {
+    #if PY_MINOR_VERSION>=11
+    LOGf << "python3.11 not supported yet";
+    return {};
+    #else
     // cnt ++;
     // if (cnt % 100 != 0) return {};
     vector<Stack> stacks;
@@ -184,6 +194,7 @@ static vector<Stack> get_stack_info() {
         }
     }
     return stacks;
+    #endif
 }
 
 template<class T>