import warnings
from torch.onnx import _CAFFE2_ATEN_FALLBACK

if not _CAFFE2_ATEN_FALLBACK:
    warnings.warn("Caffe2 support is not fully enabled in this PyTorch build. "
                  "Please enable Caffe2 by building PyTorch from source with `BUILD_CAFFE2=1` flag.")