Merge pull request #2109 from pytorch/DLFW_changes

DLFW changes
pytorch · Jul 14, 2023 · 84abc67 · 84abc67
2 parents 9cb57c2 + 267a5a2
commit 84abc67
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/examples/int8/training/vgg16/requirements.txt b/examples/int8/training/vgg16/requirements.txt
@@ -1,4 +1,3 @@
-torch>=1.10.0
 tensorboard>=1.14.0
 nvidia-pyindex
 --extra-index-url https://pypi.ngc.nvidia.com

diff --git a/py/setup.py b/py/setup.py
@@ -2,7 +2,7 @@
 import sys
 import glob
 import setuptools
-from setuptools import setup, Extension, find_packages
+from setuptools import setup, Extension, find_namespace_packages
 from setuptools.command.build_ext import build_ext
 from setuptools.command.develop import develop
 from setuptools.command.install import install
@@ -440,7 +440,7 @@ def run(self):
     },
     zip_safe=False,
     license="BSD",
-    packages=packages if FX_ONLY else find_packages(),
+    packages=packages if FX_ONLY else find_namespace_packages(),
     package_dir=package_dir if FX_ONLY else {},
     classifiers=[
         "Development Status :: 5 - Production/Stable",

diff --git a/third_party/cuda/BUILD b/third_party/cuda/BUILD
@@ -20,6 +20,9 @@ cc_library(
         ],
     }),
     hdrs = glob([
+        "include/**/*.h",
+        "include/**/*.hpp",
+        "include/**/*.inl",
         "include/**/*",
     ]),
     includes = ["include/"],
@@ -48,6 +51,9 @@ cc_library(
         ]),
     }),
     hdrs = glob([
+        "include/**/*.h",
+        "include/**/*.hpp",
+        "include/**/*.inl",
         "include/**/*",
     ]),
     includes = ["include/"],
@@ -65,6 +71,9 @@ cc_library(
         ]),
     }),
     hdrs = glob([
+        "include/**/*cublas*.h",
+        "include/**/*.hpp",
+        "include/**/*.inl",
         "include/**/*",
     ]),
     includes = ["include/"],