fixs missing include

.github/workflows/unit_test.yml

@@ -33,15 +33,15 @@ jobs:
         uses: actions/checkout@v4
       - name: Build wheel with Python 3.10
         run: |
-          python3.10 -m pip install -U setuptools poetry build six
+          python3.10 -m pip install -U setuptools poetry build six pybind11
           python3.10 -m poetry build -f wheel
       - name: Build wheel with Python 3.11
         run: |
-          python3.11 -m pip install -U setuptools poetry build six
+          python3.11 -m pip install -U setuptools poetry build six pybind11
           python3.11 -m poetry build -f wheel
       - name: Build wheel with Python 3.12
         run: |
-          python3.12 -m pip install -U setuptools poetry build six
+          python3.12 -m pip install -U setuptools poetry build six pybind11
           python3.12 -m poetry build -f wheel
       - name: Upload the wheel artifact
         uses: actions/upload-artifact@v4

pyproject.toml

@@ -24,7 +24,7 @@ exclude = [
 ]
 
 [tool.poetry.build]
-script = "build.py"
+script = "cupti_build.py"
 generate-setup-file = true
 
 [build-system]
@@ -44,7 +44,7 @@ ft_launcher = "nvidia_resiliency_ext.fault_tolerance.launcher:main"
 profile = "black"  # black-compatible
 line_length = 100  # should match black parameters
 py_version = 310  # python 3.10 as a target version
-extend_skip = ["setup.py", "build.py"]
+extend_skip = ["setup.py", "cupti_build.py"]
 
 [tool.black]
 line_length = 100

src/nvidia_resiliency_ext/straggler/cupti_src/BufferPool.cpp

@@ -19,6 +19,7 @@
 #include <cstdio>
 #include <vector>
 #include <cstdint>
+#include <cstdlib>
 #include <mutex>