Function AutoMemset

oneflow/core/kernel/kernel_util.cpp:63–68

}

void AutoMemset(ep::Stream* stream, void* dst, const char value, size_t sz,
                const MemoryCase& /*dst_mem_case*/) {
  std::unique_ptr<ep::primitive::Memset> primitive =
      ep::primitive::NewPrimitive<ep::primitive::MemsetFactory>(stream->device_type());
  primitive->Launch(stream, dst, value, sz);
}

}  //  namespace oneflow