@@ -56,7 +56,8 @@ class StubBackend final : public BackendInterface {
5656 FreeableBuffer*,
5757 ArrayRef<CompileSpec>,
5858 BackendInitContext&)>;
59- using ExecuteFn = std::function<Error(DelegateHandle*, EValue**)>;
59+ using ExecuteFn = std::function<Error(
60+ BackendExecutionContext&, DelegateHandle*, EValue**)>;
6061 using DestroyFn = std::function<void (DelegateHandle*)>;
6162
6263 // Default name that this backend is registered as.
@@ -98,7 +99,7 @@ class StubBackend final : public BackendInterface {
9899 DelegateHandle* handle,
99100 EValue** args) const override {
100101 if (execute_fn_) {
101- return execute_fn_.value ()(handle, args);
102+ return execute_fn_.value ()(context, handle, args);
102103 }
103104 // Return a benign value otherwise.
104105 return Error::Ok;
@@ -404,7 +405,9 @@ TEST_P(BackendIntegrationTest, EndToEndTestWithProcessedAsHandle) {
404405 // FreeableBuffer.
405406 DelegateHandle* execute_handle = nullptr ;
406407 StubBackend::singleton ().install_execute (
407- [&](DelegateHandle* handle, ET_UNUSED EValue** args) -> Error {
408+ [&](ET_UNUSED BackendExecutionContext& backend_execution_context,
409+ DelegateHandle* handle,
410+ ET_UNUSED EValue** args) -> Error {
408411 execute_handle = handle;
409412 auto * processed = reinterpret_cast <FreeableBuffer*>(handle);
410413
0 commit comments