We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aad03d5 commit 9aec7aeCopy full SHA for 9aec7ae
setup.py
@@ -111,17 +111,6 @@ def get_build_version():
111
sys.argv.remove("tf")
112
cfg_key = "neural_compressor_tf"
113
114
- if bool(os.getenv("USE_FP8_CONVERT", False)):
115
- from torch.utils.cpp_extension import BuildExtension, CppExtension
116
-
117
- ext_modules = [
118
- CppExtension(
119
- "fp8_convert",
120
- ["neural_compressor/torch/algorithms/habana_fp8/tensor/convert.cpp"],
121
- ),
122
- ]
123
- cmdclass = {"build_ext": BuildExtension}
124
125
project_name = PKG_INSTALL_CFG[cfg_key].get("project_name")
126
include_packages = PKG_INSTALL_CFG[cfg_key].get("include_packages") or {}
127
package_data = PKG_INSTALL_CFG[cfg_key].get("package_data") or {}
0 commit comments