diff --git a/aten/src/ATen/core/op_registration/op_whitelist.h b/aten/src/ATen/core/op_registration/op_whitelist.h index cf7f09bb0f9..c8437e924a3 100644 --- a/aten/src/ATen/core/op_registration/op_whitelist.h +++ b/aten/src/ATen/core/op_registration/op_whitelist.h @@ -1,5 +1,10 @@ #pragma once +// TODO: unify to C10_MOBILE. In theory this header could be used in OSS. +#ifdef XPLAT_MOBILE_BUILD +#include +#endif + /** * This header implements functionality to build PyTorch with only a certain * set of operators (+ dependencies) included.