MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MemZero

Method MemZero

tensorflow/stream_executor/host/host_gpu_executor.cc:120–128  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

118}
119
120port::Status HostExecutor::MemZero(Stream *stream, DeviceMemoryBase *location,
121 uint64 size) {
122 void *gpu_mem = location->opaque();
123 // Enqueue the [asynchronous] memzero on the stream (HostStream) associated
124 // with the HostExecutor.
125 AsHostStream(stream)->EnqueueTask(
126 [gpu_mem, size]() { memset(gpu_mem, 0, size); });
127 return port::Status::OK();
128}
129
130port::Status HostExecutor::Memset(Stream *stream, DeviceMemoryBase *location,
131 uint8 pattern, uint64 size) {

Callers

nothing calls this directly

Calls 3

AsHostStreamFunction · 0.85
opaqueMethod · 0.80
EnqueueTaskMethod · 0.45

Tested by

no test coverage detected