From f8eb799aee047fda46723dc8bf323bc5e552d8fd Mon Sep 17 00:00:00 2001 From: Emmanuel Ferdman Date: Sun, 6 Oct 2024 18:18:54 +0300 Subject: [PATCH] [PyTorch] remove duplicate code (#1215) Signed-off-by: Emmanuel Ferdman Co-authored-by: Kirthi Shankar Sivamani --- tests/pytorch/fused_attn/test_fused_attn.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/tests/pytorch/fused_attn/test_fused_attn.py b/tests/pytorch/fused_attn/test_fused_attn.py index d110dece53..fa371a38bb 100644 --- a/tests/pytorch/fused_attn/test_fused_attn.py +++ b/tests/pytorch/fused_attn/test_fused_attn.py @@ -1854,13 +1854,6 @@ def get_dummy_cuda_rng_tracker() -> CudaRNGStatesTracker: """Get cuda rng tracker.""" return _DUMMY_CUDA_RNG_STATE_TRACKER - _DUMMY_CUDA_RNG_STATE_TRACKER = CudaRNGStatesTracker() - _DUMMY_CUDA_RNG_STATE_TRACKER.add("model-parallel-rng", seed) - - def get_dummy_cuda_rng_tracker(): - """Get cuda rng tracker.""" - return _DUMMY_CUDA_RNG_STATE_TRACKER - block = DotProductAttention( config.num_heads, config.head_dim_qk,