(D, dtype, device, is_causal, is_fp8=False)
| 143 | @pytest.mark.parametrize("device", ["cuda"]) |
| 144 | @pytest.mark.parametrize("is_causal", [False, True]) |
| 145 | def test_benchmark_attn_func(D, dtype, device, is_causal, is_fp8=False): |
| 146 | _test_benchmark_attn_func(D, dtype, device, is_causal, is_fp8) |
| 147 | |
| 148 | |
| 149 | @pytest.mark.parametrize("D", [64, 128, 256]) |
nothing calls this directly
no test coverage detected