(random_tensor, mask_value)
| 241 | |
| 242 | @pytest.mark.parametrize("mask_value", [100, 200, 300]) |
| 243 | def test_mask_persistency(random_tensor, mask_value): |
| 244 | block = ParallelTransformerBlock(512, 64, 8, 4) |
| 245 | block.get_mask(mask_value, random_tensor.device) |
| 246 | assert block.mask.shape[0] == mask_value |
| 247 | |
| 248 | |
| 249 | @pytest.mark.parametrize( |
nothing calls this directly
no outgoing calls
no test coverage detected