From 94e4e87b8876f1cdec07fc36c0de02cff58698f1 Mon Sep 17 00:00:00 2001
From: foreman
Date: Wed, 4 Dec 2019 13:34:03 -0500
Subject: [PATCH] P4 to Git Change 2040867 by kjayapra@0_HIPWS_LNX1_ROCM on
2019/12/04 13:32:27
SWDEV-145570 - Adding back the lazy kernel changes because the OOM issue is because of KFD/RocR.
Affected files ...
... //depot/stg/opencl/drivers/opencl/api/hip/hip_platform.cpp#53 edit
---
hipamd/api/hip/hip_platform.cpp | 3 +++
1 file changed, 3 insertions(+)
diff --git a/hipamd/api/hip/hip_platform.cpp b/hipamd/api/hip/hip_platform.cpp
index df2107974e..37d022faa9 100644
--- a/hipamd/api/hip/hip_platform.cpp
+++ b/hipamd/api/hip/hip_platform.cpp
@@ -416,6 +416,9 @@ extern "C" void __hipRegisterFunction(
HIP_INIT();
PlatformState::DeviceFunction func{ std::string{deviceName}, modules, std::vector{ g_devices.size() }};
PlatformState::instance().registerFunction(hostFunction, func);
+ for (size_t i = 0; i < g_devices.size(); ++i) {
+ PlatformState::instance().getFunc(hostFunction, i);
+ }
}
// Registers a device-side global variable.