Adding PyProf to Apex (#404)

Co-authored-by: Aditya Agrawal <aditya.iitb@gmail.com>
Co-authored-by: Marek Kolodziej <mkolod@gmail.com>

requirements.txt

+cxxfilt>=0.2.0
+tqdm>=4.28.1
+numpy>=1.15.3
+PyYAML>=5.1
+pytest>=3.5.1

setup.py

@@ -2,7 +2,9 @@ import torch
 from setuptools import setup, find_packages
 import subprocess
 
+from pip._internal import main as pipmain
 import sys
+import warnings
 
 if not torch.cuda.is_available():
     print("\nWarning: Torch did not find available GPUs on this system.\n",
@@ -19,6 +21,17 @@ if TORCH_MAJOR == 0 and TORCH_MINOR < 4:
 cmdclass = {}
 ext_modules = []
 
+if "--pyprof" in sys.argv:
+    with open('requirements.txt') as f:
+        required_packages = f.read().splitlines()
+        pipmain(["install"] + required_packages)
+    try:
+        sys.argv.remove("--pyprof")
+    except:
+        pass
+else:
+    warnings.warn("Option --pyprof not specified. Not installing PyProf dependencies!")
+
 if "--cpp_ext" in sys.argv or "--cuda_ext" in sys.argv:
     if TORCH_MAJOR == 0:
         raise RuntimeError("--cpp_ext requires Pytorch 1.0 or later, "

tests/L0/run_pyprof_nvtx/__init__.py

+import test_pyprof_nvtx.TestPyProfNvtx as TestPyProfNvtx