Spaces:
Running
on
Zero
Running
on
Zero
File size: 537 Bytes
db6a3b7 690b53e db6a3b7 690b53e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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 *
|