From a820df6bbd522a62eb207134815415fb1a2a4bb8 Mon Sep 17 00:00:00 2001 From: Kai Li Date: Thu, 20 Nov 2025 10:18:58 -0800 Subject: [PATCH] Add is_fbcode() check to unblock unrecognized torchao version (#3338) Summary: As title. Differential Revision: D87008228 --- torchao/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/torchao/__init__.py b/torchao/__init__.py index b0baf6ff4a..a172674109 100644 --- a/torchao/__init__.py +++ b/torchao/__init__.py @@ -26,6 +26,10 @@ logger = logging.getLogger(__name__) +def is_fbcode(): + return not hasattr(torch.version, "git_version") + + def _parse_version(version_string): """ Parse version string representing pre-release with -1 @@ -48,7 +52,7 @@ def _parse_version(version_string): force_skip_loading_so_files = ( os.getenv("TORCHAO_FORCE_SKIP_LOADING_SO_FILES", "0") == "1" ) -if force_skip_loading_so_files: +if force_skip_loading_so_files or is_fbcode(): # user override # users can set env var TORCHAO_FORCE_SKIP_LOADING_SO_FILES=1 to skip loading .so files # this way, if they are using an incompatbile torch version, they can still use the API by setting the env var