Method get_max

imperative/python/test/unit/core/test_stream.py:21–22

(self, device)

            mge.device.reset_max_memory_stats(d)

    def get_max(self, device):
        return mge.device.get_max_allocated_memory(device) - self.baseline[device]


@pytest.mark.require_ngpu(1)