Skip to content

Commit c967b70

Browse files
authored
Add timeout default in NeutronBackend (pytorch#21293)
Differential Revision: D113453148 Pull Request resolved: pytorch#21293
1 parent 69014b6 commit c967b70

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

backends/nxp/runtime/NeutronBackend.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -290,6 +290,10 @@ class NeutronBackend final : public PyTorchBackendInterface {
290290

291291
auto* cfg = allocator->allocateInstance<NeutronExecutorchConfig>();
292292

293+
// allocateInstance returns raw, uninitialized memory (no constructor runs),
294+
// so set the driver-config timeout explicitly.
295+
cfg->mcfg.timeoutSeconds = 60;
296+
293297
// The following data is read from the "processed" data blob.
294298
// cfg->numInputs
295299
// cfg->numoutputs
@@ -673,4 +677,4 @@ static auto registered = register_backend(backend_id);
673677
} // namespace
674678
} // namespace neutron
675679
} // namespace executor
676-
} // namespace torch
680+
} // namespace torch

0 commit comments

Comments
 (0)