remove input_ids from ForwardMeta (#4793)
CE Compile Job / ce_job_pre_check (push) Has been cancelled
CE Compile Job / print_ce_job_pre_check_outputs (push) Has been cancelled
CE Compile Job / FD-Clone-Linux (push) Has been cancelled
CE Compile Job / Show Code Archive Output (push) Has been cancelled
CE Compile Job / BUILD_SM8090 (push) Has been cancelled
CE Compile Job / BUILD_SM8689 (push) Has been cancelled
CE Compile Job / CE_UPLOAD (push) Has been cancelled
Deploy GitHub Pages / deploy (push) Has been cancelled

This commit is contained in:
周周周
2025-11-05 11:55:51 +08:00
committed by GitHub
parent 9676cc87d6
commit 876e4a8935
13 changed files with 26 additions and 41 deletions
@@ -108,9 +108,7 @@ class TestGraphOptBackend(unittest.TestCase):
# Create input data
self.input_tensor = paddle.randint(32, shape=self.input_shape, dtype=self.dtype)
self.forward_meta = ForwardMeta(
input_ids=self.input_tensor, ids_remove_padding=self.input_tensor, step_use_cudagraph=True
)
self.forward_meta = ForwardMeta(ids_remove_padding=self.input_tensor, step_use_cudagraph=True)
# Compute baseline result once
baseline_model = Attention(fd_config=self.baseline_fd_config, **self.model_config)