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

Method Memset32

tensorflow/stream_executor/rocm/rocm_gpu_executor.cc:562–572  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

560}
561
562port::Status GpuExecutor::Memset32(Stream* stream, DeviceMemoryBase* location,
563 uint32 pattern, uint64 size) {
564 VLOG(2) << "enqueueing memset32 operation onto stream " << stream
565 << " at location " << location << " with size " << size
566 << " and pattern " << std::hex << pattern;
567 CHECK(reinterpret_cast<uintptr_t>(location->opaque()) % 4 == 0 &&
568 size % 4 == 0);
569 return GpuDriver::AsynchronousMemsetUint32(
570 context_, AsROCmDevicePtr(location), pattern, size / 4,
571 AsGpuStreamValue(stream));
572}
573
574bool GpuExecutor::Memcpy(Stream* stream, void* host_dst,
575 const DeviceMemoryBase& gpu_src, uint64 size) {

Callers

nothing calls this directly

Calls 3

AsROCmDevicePtrFunction · 0.85
AsGpuStreamValueFunction · 0.85
opaqueMethod · 0.80

Tested by

no test coverage detected