Skip to content

Commit 1aadcc0

Browse files
cccclaifacebook-github-bot
authored andcommitted
1 parent 545535b commit 1aadcc0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

runtime/executor/test/backend_integration_test.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ 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(BackendExecutionContext&, DelegateHandle*, EValue**)>;
6060
using DestroyFn = std::function<void(DelegateHandle*)>;
6161

6262
// Default name that this backend is registered as.
@@ -98,7 +98,7 @@ class StubBackend final : public BackendInterface {
9898
DelegateHandle* handle,
9999
EValue** args) const override {
100100
if (execute_fn_) {
101-
return execute_fn_.value()(handle, args);
101+
return execute_fn_.value()(context, handle, args);
102102
}
103103
// Return a benign value otherwise.
104104
return Error::Ok;
@@ -404,7 +404,7 @@ TEST_P(BackendIntegrationTest, EndToEndTestWithProcessedAsHandle) {
404404
// FreeableBuffer.
405405
DelegateHandle* execute_handle = nullptr;
406406
StubBackend::singleton().install_execute(
407-
[&](DelegateHandle* handle, ET_UNUSED EValue** args) -> Error {
407+
[&](ET_UNUSED BackendExecutionContext& backend_execution_context, DelegateHandle* handle, ET_UNUSED EValue** args) -> Error {
408408
execute_handle = handle;
409409
auto* processed = reinterpret_cast<FreeableBuffer*>(handle);
410410

0 commit comments

Comments
 (0)