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): """ 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):