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

Function SyncAutoMemcpy

oneflow/core/kernel/kernel_util.cpp:57–61  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

55}
56
57void SyncAutoMemcpy(ep::Stream* stream, void* dst, const void* src, size_t sz,
58 const MemoryCase& dst_mem_case, const MemoryCase& src_mem_case) {
59 AutoMemcpy(stream, dst, src, sz, dst_mem_case, src_mem_case);
60 CHECK_JUST(stream->Sync());
61}
62
63void AutoMemset(ep::Stream* stream, void* dst, const char value, size_t sz,
64 const MemoryCase& /*dst_mem_case*/) {

Callers 14

CopyLocalTensorDataToFunction · 0.85
ComputeMethod · 0.85
GetTensorItemValueFunction · 0.85
_rnn_pack_sequence_implFunction · 0.85
_lstm_pack_sequence_implFunction · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
CopyFromNumpyArrayFunction · 0.85

Calls 2

AutoMemcpyFunction · 0.85
SyncMethod · 0.45

Tested by

no test coverage detected