From 6300b0908fc48acff5b6f244daf6b727307b72e1 Mon Sep 17 00:00:00 2001 From: Dun Liang Date: Fri, 1 Oct 2021 19:56:47 +0800 Subject: [PATCH] lower arch support for 30 --- python/jittor/__init__.py | 2 +- python/jittor/compiler.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/jittor/__init__.py b/python/jittor/__init__.py index 0fbbe3b2..424644b5 100644 --- a/python/jittor/__init__.py +++ b/python/jittor/__init__.py @@ -9,7 +9,7 @@ # file 'LICENSE.txt', which is part of this source code package. # *************************************************************** -__version__ = '1.3.0.2' +__version__ = '1.3.0.3' from jittor_utils import lock with lock.lock_scope(): ori_int = int diff --git a/python/jittor/compiler.py b/python/jittor/compiler.py index 34e6b3fb..22b4f580 100644 --- a/python/jittor/compiler.py +++ b/python/jittor/compiler.py @@ -1283,7 +1283,7 @@ if has_cuda: archs = [] for arch in flags.cuda_archs: if arch<50: - LOG.w(f"CUDA arch({arch})<50 is not supported") + LOG.w(f"CUDA arch({arch})<30 is not supported") continue archs.append(arch) nvcc_flags += f" -arch=compute_{min(flags.cuda_archs)} "