Profile refine ()

* fix profile api high version test=develop
revert-21172-masked_select_api
wangchaochaohu 5 years ago committed by GitHub
parent 67c836fb5c
commit 8293f21a52
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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);

Loading…
Cancel
Save