MegEngine/imperative/python/megengine/amp/__init__.py

# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import mprop

from ..core.tensor.amp import *
from .autocast import autocast
from .grad_scaler import GradScaler

mprop.init()