fix CI build

spconv/__init__.py

@@ -14,5 +14,5 @@
 
 from . import build as _build
 
-from .algo import ConvAlgo
-from . import utils, constants
\ No newline at end of file
+from .core import ConvAlgo, AlgoHint
+from . import constants
\ No newline at end of file

spconv/algo.py

@@ -23,20 +23,7 @@ from typing import Optional
 import time
 
 import numpy as np
-
-
-class ConvAlgo(Enum):
-    Native = "Native"
-    MaskImplicitGemm = "MaskImplicitGemm"
-    MaskSplitImplicitGemm = "MaskSplitImplicitGemm"
-
-
-class AlgoHint(Enum):
-    NoHint = 0b000
-    Fowrard = 0b001
-    BackwardInput = 0b010
-    BackwardWeight = 0b100
-
+from .core import ConvAlgo, AlgoHint
 
 ALL_ALGO_DESPS = GemmMainUnitTest.get_all_algo_desp()
 

spconv/core.py

+# Copyright 2021 Yan Yan
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from enum import Enum
+
+class ConvAlgo(Enum):
+    Native = "Native"
+    MaskImplicitGemm = "MaskImplicitGemm"
+    MaskSplitImplicitGemm = "MaskSplitImplicitGemm"
+
+
+class AlgoHint(Enum):
+    NoHint = 0b000
+    Fowrard = 0b001
+    BackwardInput = 0b010
+    BackwardWeight = 0b100
+

tools/build-wheels.sh

@@ -28,7 +28,7 @@ function repair_wheel {
 export SPCONV_DISABLE_JIT="1"
 export CUMM_CUDA_ARCH_LIST="all"
 # Compile wheels, we only support 3.6-3.10.
-"/opt/python/cp36-cp36m/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp
+# "/opt/python/cp36-cp36m/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp
 "/opt/python/cp37-cp37m/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp
 "/opt/python/cp38-cp38/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp
 "/opt/python/cp39-cp39/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp