Skip to content

Commit f8eb799

Browse files
[PyTorch] remove duplicate code (#1215)
Signed-off-by: Emmanuel Ferdman <[email protected]> Co-authored-by: Kirthi Shankar Sivamani <[email protected]>
1 parent 9d976bc commit f8eb799

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

tests/pytorch/fused_attn/test_fused_attn.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1854,13 +1854,6 @@ def get_dummy_cuda_rng_tracker() -> CudaRNGStatesTracker:
18541854
"""Get cuda rng tracker."""
18551855
return _DUMMY_CUDA_RNG_STATE_TRACKER
18561856

1857-
_DUMMY_CUDA_RNG_STATE_TRACKER = CudaRNGStatesTracker()
1858-
_DUMMY_CUDA_RNG_STATE_TRACKER.add("model-parallel-rng", seed)
1859-
1860-
def get_dummy_cuda_rng_tracker():
1861-
"""Get cuda rng tracker."""
1862-
return _DUMMY_CUDA_RNG_STATE_TRACKER
1863-
18641857
block = DotProductAttention(
18651858
config.num_heads,
18661859
config.head_dim_qk,

0 commit comments

Comments
 (0)