Skip to content

Commit d5127ab

Browse files
committed
fix build error
1 parent 59f8598 commit d5127ab

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/fluid/platform/device_context_test_cuda_graph.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ TEST(Device, DeviceContextWithCUDAGraph) {
3131
auto* dev_ctx = pool.Get(place);
3232

3333
paddle::platform::BeginCUDAGraphCapture(
34-
place, gpuStreamCaptureMode::cudaStreamCaptureModeThreadLocal, 0);
34+
place, cudaStreamCaptureMode::cudaStreamCaptureModeThreadLocal, 0);
3535
ASSERT_EQ(dev_ctx->IsCUDAGraphAllocatorValid(), true);
3636
dev_ctx->GetCUDAGraphAllocator();
3737
paddle::platform::EndCUDAGraphCapture();

paddle/phi/backends/gpu/rocm/hip_graph.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ void CUDAGraph::BeginSegmentCapture() {
160160

161161
void CUDAGraph::BeginCapture(phi::GPUPlace place,
162162
gpuStream_t stream,
163-
gpuStreamCaptureMode mode) {
163+
hipStreamCaptureMode mode) {
164164
ThrowErrorIfNotSupportCUDAGraph();
165165
#if defined(PADDLE_WITH_HIP)
166166
PADDLE_ENFORCE_EQ(IsCapturing(),

0 commit comments

Comments
 (0)