MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / SoftSyncNNGraphBuffers

Function SoftSyncNNGraphBuffers

oneflow/core/framework/nn_graph.cpp:915–923  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

913}
914
915Maybe<void> SoftSyncNNGraphBuffers(const one::TensorTuple& buffers,
916 const std::shared_ptr<NNGraph>& nn_graph) {
917 const auto& eager_blob_objects = std::make_shared<vm::EagerBlobObjectList>();
918 JUST(MakeEagerBlobObjectList(eager_blob_objects.get(), buffers));
919 JUST(PhysicalRun([&](InstructionsBuilder* builder) -> Maybe<void> {
920 return builder->SoftSyncNNGraphBuffers(eager_blob_objects, nn_graph);
921 }));
922 return Maybe<void>::Ok();
923}
924
925} // namespace oneflow

Callers 2

LaunchLazyJobMethod · 0.85
RunMethod · 0.85

Calls 4

MakeEagerBlobObjectListFunction · 0.85
PhysicalRunFunction · 0.85
getMethod · 0.45

Tested by

no test coverage detected