Initial commit

pkgs/xformers/profiler/__init__.py

@@ -0,0 +1,17 @@
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
+#
+# This source code is licensed under the BSD license found in the
+# LICENSE file in the root directory of this source tree.
+
+from .api import profile, step
+from .profiler import MemSnapshotsProfiler, NsightProfiler, PyTorchProfiler
+from .slow_ops_profiler import DetectSlowOpsProfiler
+
+__all__ = [
+    "profile",
+    "step",
+    "MemSnapshotsProfiler",
+    "PyTorchProfiler",
+    "NsightProfiler",
+    "DetectSlowOpsProfiler",
+]