Fix Module 15 memoization: Add optional mask parameter to MockTransformerBlock forward method

This commit is contained in:
Vijay Janapa Reddi
2025-11-10 07:26:11 -05:00
parent 6a409bab19
commit d73e1e9eed

View File

@@ -1281,7 +1281,7 @@ def test_unit_noninvasive_integration():
def __init__(self):
self.attention = self
def forward(self, x):
def forward(self, x, mask=None):
# Simple pass-through for testing
return x