restore

PaddlePaddle · Apr 23, 2024 · 9844a3b · 9844a3b
1 parent 375c415
commit 9844a3b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/phi/api/profiler/device_tracer.cc b/paddle/phi/api/profiler/device_tracer.cc
@@ -176,9 +176,9 @@ void DisableActivity() {
 void CUPTIAPI bufferRequested(uint8_t **buffer,
                               size_t *size,
                               size_t *maxNumRecords) {
-  std::vector<uint8_t> buf(kBufSize + kAlignSize);
+  uint8_t *buf = reinterpret_cast<uint8_t *>(malloc(kBufSize + kAlignSize));
   *size = kBufSize;
-  *buffer = ALIGN_BUFFER(buf.data(), kAlignSize);
+  *buffer = ALIGN_BUFFER(buf, kAlignSize);
   *maxNumRecords = 0;
 }