souging/TRELLIS_TextTo3D

1from .. import BACKEND2 3 4SPCONV_ALGO = 'auto'    # 'auto', 'implicit_gemm', 'native'5 6def __from_env():7    import os8        9    global SPCONV_ALGO10    env_spconv_algo = os.environ.get('SPCONV_ALGO')11    if env_spconv_algo is not None and env_spconv_algo in ['auto', 'implicit_gemm', 'native']:12        SPCONV_ALGO = env_spconv_algo13    print(f"[SPARSE][CONV] spconv algo: {SPCONV_ALGO}")14        15 16__from_env()17 18if BACKEND == 'torchsparse':19    from .conv_torchsparse import *20elif BACKEND == 'spconv':21    from .conv_spconv import *22