envs.py 648 Bytes
Newer Older
1
import os
gushiqiao's avatar
gushiqiao committed
2
import torch
helloyongyang's avatar
helloyongyang committed
3
from functools import lru_cache
4
5


helloyongyang's avatar
helloyongyang committed
6
7
8
9
@lru_cache(maxsize=None)
def CHECK_ENABLE_PROFILING_DEBUG():
    ENABLE_PROFILING_DEBUG = os.getenv("ENABLE_PROFILING_DEBUG", "false").lower() == "true"
    return ENABLE_PROFILING_DEBUG
10

helloyongyang's avatar
helloyongyang committed
11
12
13
14
15

@lru_cache(maxsize=None)
def CHECK_ENABLE_GRAPH_MODE():
    ENABLE_GRAPH_MODE = os.getenv("ENABLE_GRAPH_MODE", "false").lower() == "true"
    return ENABLE_GRAPH_MODE
16
17
18
19
20
21


@lru_cache(maxsize=None)
def GET_RUNNING_FLAG():
    RUNNING_FLAG = os.getenv("RUNNING_FLAG", "infer")
    return RUNNING_FLAG
gushiqiao's avatar
gushiqiao committed
22
23
24
25
26
27


@lru_cache(maxsize=None)
def GET_DTYPE():
    RUNNING_FLAG = os.getenv("DTYPE")
    return RUNNING_FLAG