From 8293f21a527e1521189a0a5f167bb873a76da0d0 Mon Sep 17 00:00:00 2001
From: wangchaochaohu <wangchao66@baidu.com>
Date: Thu, 28 Nov 2019 17:12:19 +0800
Subject: [PATCH] Profile refine (#21258)

* fix profile api high version test=develop
---
 paddle/fluid/platform/device_tracer.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/paddle/fluid/platform/device_tracer.cc b/paddle/fluid/platform/device_tracer.cc
index 8458b17f82..9aa6aa44b2 100644
--- a/paddle/fluid/platform/device_tracer.cc
+++ b/paddle/fluid/platform/device_tracer.cc
@@ -185,8 +185,13 @@ void CUPTIAPI bufferCompleted(CUcontext ctx, uint32_t streamId, uint8_t *buffer,
         switch (record->kind) {
           case CUPTI_ACTIVITY_KIND_KERNEL:
           case CUPTI_ACTIVITY_KIND_CONCURRENT_KERNEL: {
+#if CUDA_VERSION >= 9000
+            auto *kernel =
+                reinterpret_cast<const CUpti_ActivityKernel4 *>(record);
+#else
             auto *kernel =
                 reinterpret_cast<const CUpti_ActivityKernel3 *>(record);
+#endif
             tracer->AddKernelRecords(kernel->name, kernel->start, kernel->end,
                                      kernel->deviceId, kernel->streamId,
                                      kernel->correlationId);
-- 
GitLab