Add convenience import (#51195)

Summary: Pull Request resolved: #51195 Add kineto_available to torch.profiler Test Plan: >>> import torch.profiler >>> torch.profiler.kineto_available() True Reviewed By: ngimel Differential Revision: D26113906 Pulled By: ilia-cher fbshipit-source-id: fe4502d29d10d8bd9459b0504aa0ee856af43acc
pytorch · Jan 28, 2021 · d14d8c7 · d14d8c7
1 parent ea0d304
commit d14d8c7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/profiler/__init__.py b/torch/profiler/__init__.py
@@ -10,3 +10,4 @@
 '''
 
 from .profiler import profile, schedule, ProfilerAction, ProfilerActivity
+from torch.autograd import kineto_available