File size: 551 Bytes
db6a3b7 690b53e a6bbecf 690b53e a6bbecf 690b53e a6bbecf 690b53e a6bbecf 690b53e a6bbecf db6a3b7 a6bbecf 690b53e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from .. import BACKEND
SPCONV_ALGO = "auto" # 'auto', 'implicit_gemm', 'native'
def __from_env():
import os
global SPCONV_ALGO
env_spconv_algo = os.environ.get("SPCONV_ALGO")
if env_spconv_algo is not None and env_spconv_algo in [
"auto",
"implicit_gemm",
"native",
]:
SPCONV_ALGO = env_spconv_algo
print(f"[SPARSE][CONV] spconv algo: {SPCONV_ALGO}")
__from_env()
if BACKEND == "torchsparse":
from .conv_torchsparse import *
elif BACKEND == "spconv":
from .conv_spconv import *
|