From 5b29be8f9d9acfe9855e11473754d55bae6c2f0b Mon Sep 17 00:00:00 2001 From: root <26priya11@gmail.com> Date: Tue, 29 Oct 2024 15:29:50 -0700 Subject: [PATCH] remove deprecated clear cuda cache --- benchmarks/python/test_adaptive_layernorm_host.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/benchmarks/python/test_adaptive_layernorm_host.py b/benchmarks/python/test_adaptive_layernorm_host.py index 7e3c67b6d8b..7e60da7659d 100644 --- a/benchmarks/python/test_adaptive_layernorm_host.py +++ b/benchmarks/python/test_adaptive_layernorm_host.py @@ -3,7 +3,6 @@ # SPDX-License-Identifier: BSD-3-Clause import pytest from nvfuser import FusionDefinition, DataType -from nvfuser.pytorch_utils import clear_cuda_cache from .core import run_benchmark import torch @@ -73,8 +72,6 @@ def test_adaptive_layernorm_fwd_benchmark( disable_validation: bool, disable_benchmarking: bool, ): - clear_cuda_cache() - B = 1 T = 30 * 1024 D = 1024