We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9d976bc commit f8eb799Copy full SHA for f8eb799
tests/pytorch/fused_attn/test_fused_attn.py
@@ -1854,13 +1854,6 @@ def get_dummy_cuda_rng_tracker() -> CudaRNGStatesTracker:
1854
"""Get cuda rng tracker."""
1855
return _DUMMY_CUDA_RNG_STATE_TRACKER
1856
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
1864
block = DotProductAttention(
1865
config.num_heads,
1866
config.head_dim_qk,
0 commit comments