From 7703031d33d2111ba33cc159aa4eeefd4dce7e9c Mon Sep 17 00:00:00 2001
From: mhbliao <47895780+mhbliao@users.noreply.github.com>
Date: Fri, 6 Dec 2019 12:49:17 -0500
Subject: [PATCH] Fix `hipExtLaunchMultiKernelMultiDevice` refactoring. (#1714)

- Use the correct condition for HIP VDI runtime.

[ROCm/hip commit: e9da934ac6287dc24523d8278f61141a2e928235]
---
 projects/hip/include/hip/hcc_detail/hip_runtime_api.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/hip/include/hip/hcc_detail/hip_runtime_api.h b/projects/hip/include/hip/hcc_detail/hip_runtime_api.h
index 4397e7fe50..e38c3a2dd7 100644
--- a/projects/hip/include/hip/hcc_detail/hip_runtime_api.h
+++ b/projects/hip/include/hip/hcc_detail/hip_runtime_api.h
@@ -2956,7 +2956,7 @@ hipError_t hipOccupancyMaxActiveBlocksPerMultiprocessor(
 hipError_t hipOccupancyMaxActiveBlocksPerMultiprocessorWithFlags(
    uint32_t* numBlocks, hipFunction_t f, uint32_t blockSize, size_t dynSharedMemPerBlk, unsigned int flags);
 
-#if defined(__clang__) && defined(__HIP__)
+#if __HIP_VDI__
 /**
  * @brief Launches kernels on multiple devices and guarantees all specified kernels are dispatched
  * on respective streams before enqueuing any other work on the specified streams from any other threads