source.privacy feature

ghstack-source-id: 4c40af24a6b11983523033590e0959bf39dc9d3c Pull Request resolved: fairinternal/xformers#428 __original_commit__ = fairinternal/xformers@36fb05d
P2Enjoy · Jan 15, 2023 · 6f3c20f · 6f3c20f
1 parent 5db2259
commit 6f3c20f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/xformers/__init__.py b/xformers/__init__.py
@@ -21,6 +21,7 @@
 
 # Set to true to utilize functorch
 _is_functorch_available: bool = False
+_is_opensource: bool = True
 
 
 def compute_once(func):

diff --git a/xformers/info.py b/xformers/info.py
@@ -8,7 +8,14 @@
 
 import torch
 
-from . import __version__, _cpp_lib, _is_functorch_available, _is_triton_available, ops
+from . import (
+    __version__,
+    _cpp_lib,
+    _is_functorch_available,
+    _is_opensource,
+    _is_triton_available,
+    ops,
+)
 from .ops.common import OPERATORS_REGISTRY
 
 
@@ -51,6 +58,11 @@ def print_info():
     else:
         features["build.info"] = "none"
 
+    if _is_opensource:
+        features["source.privacy"] = "open source"
+    else:
+        features["source.privacy"] = "fairinternal"
+
     for name, status in features.items():
         print("{:<50} {}".format(f"{name}:", status))