From 320ae29a01f818d0d5bfa483a2695449a81d22fe Mon Sep 17 00:00:00 2001 From: Rick Ratzel Date: Fri, 12 May 2023 10:57:04 -0500 Subject: [PATCH 1/2] Fixed bug for handling self.gpuData being empty. --- .../rapids_pytest_benchmark/plugin.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/rapids_pytest_benchmark/rapids_pytest_benchmark/plugin.py b/rapids_pytest_benchmark/rapids_pytest_benchmark/plugin.py index 43f3e7b..fd6eac3 100644 --- a/rapids_pytest_benchmark/rapids_pytest_benchmark/plugin.py +++ b/rapids_pytest_benchmark/rapids_pytest_benchmark/plugin.py @@ -210,15 +210,24 @@ def gpu_rounds(self): @pytest_benchmark_utils.cached_property def gpu_mem(self): - return max([i[0] for i in self.gpuData]) + if len(self.gpuData) > 0: + return max([i[0] for i in self.gpuData]) + else: + return 0 @pytest_benchmark_utils.cached_property def gpu_util(self): - return max([i[1] for i in self.gpuData]) + if len(self.gpuData) > 0: + return max([i[1] for i in self.gpuData]) + else: + return 0 @pytest_benchmark_utils.cached_property def gpu_leaked_mem(self): - return max([i[2] for i in self.gpuData]) + if len(self.gpuData) > 0: + return max([i[2] for i in self.gpuData]) + else: + return 0 class GPUBenchmarkFixture(pytest_benchmark_fixture.BenchmarkFixture): From d4ecae0cbf2500062c23b05a1aabfdd212f92caa Mon Sep 17 00:00:00 2001 From: Rick Ratzel Date: Fri, 12 May 2023 11:08:30 -0500 Subject: [PATCH 2/2] Bumps version. --- rapids_pytest_benchmark/rapids_pytest_benchmark/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rapids_pytest_benchmark/rapids_pytest_benchmark/__init__.py b/rapids_pytest_benchmark/rapids_pytest_benchmark/__init__.py index a09f519..6707045 100644 --- a/rapids_pytest_benchmark/rapids_pytest_benchmark/__init__.py +++ b/rapids_pytest_benchmark/rapids_pytest_benchmark/__init__.py @@ -11,7 +11,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -__version__ = "0.0.15" +__version__ = "0.0.16" def setFixtureParamNames(request, orderedParamNameList): """