()
| 4 | SPCONV_ALGO = 'auto' # 'auto', 'implicit_gemm', 'native' |
| 5 | |
| 6 | def __from_env(): |
| 7 | import os |
| 8 | |
| 9 | global SPCONV_ALGO |
| 10 | 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_algo |
| 13 | print(f"[SPARSE][CONV] spconv algo: {SPCONV_ALGO}") |
| 14 | |
| 15 | |
| 16 | __from_env() |