env.py 328 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10
"""Global configuration/variable scope for autotvm"""

class AutotvmGlobalScope(object):
    current = None

    def __init__(self):
        self._old = AutotvmGlobalScope.current
        AutotvmGlobalScope.current = self

        self.cuda_target_arch = None
11
        self.in_tuning = False
12 13

GLOBAL_SCOPE = AutotvmGlobalScope()