Use get_include()

mlcommons · Sep 26, 2023 · ff530c0 · ff530c0
1 parent 75dd4e4
commit ff530c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/loadgen/setup.py b/loadgen/setup.py
@@ -77,7 +77,7 @@
 mlperf_loadgen_module = Pybind11Extension(
         "mlperf_loadgen",
         define_macros=[("MAJOR_VERSION", "3"), ("MINOR_VERSION", "1")],
-        include_dirs=[".", "../third_party/pybind/include"],
+        include_dirs=[".", get_include()],
         sources=mlperf_loadgen_sources,
         depends=mlperf_loadgen_headers)