mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
e07134092f
Confusing backtraces are issued to users when they run Caffe2 scripts (or tests) on PyTorch builds without Caffe2 enabled through `BUILD_CAFFE2=1` This PR adds warnings (in more than one place) to return a friendly message for the user, helping them to overcome the problem by themselves Pull Request resolved: https://github.com/pytorch/pytorch/pull/73770 Approved by: https://github.com/BowenBao, https://github.com/malfet, https://github.com/garymm
7 lines
274 B
Python
7 lines
274 B
Python
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.")
|