Skip to content

GitLab

Unverified Commit 25bfcb91 authored by Masaki Kozuki's avatar Masaki Kozuki Committed by GitHub
Browse files

conditionally import amp_C (#1211)

parent 2205cff2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
apex/transformer/pipeline_parallel/utils.py
View file @ 25bfcb91
......@@ -20,10 +20,11 @@ import torch
from torch.nn.parallel import DistributedDataParallel
from apex.multi_tensor_apply import multi_tensor_applier
import amp_C
from apex.transformer import parallel_state
from apex.transformer.microbatches import build_num_microbatches_calculator
from apex.transformer.pipeline_parallel._timers import _Timers
if multi_tensor_applier.available:
import amp_C
_GLOBAL_ARGS = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment