Module neuralops.tests.test_deterministic

Tests for deterministic attention mode.

Functions

def test_deterministic_attention_toggle()

Expand source code

def test_deterministic_attention_toggle():
    assert not is_deterministic_attention()
    prev = enable_deterministic_attention(True)
    assert not prev
    assert is_deterministic_attention()
    prev2 = enable_deterministic_attention(False)
    assert prev2
    assert not is_deterministic_attention()