diff --git a/lib/Runtime/ROCm/FlyRocmRuntimeWrappers.cpp b/lib/Runtime/ROCm/FlyRocmRuntimeWrappers.cpp
index a06ce9a34..cd62b2668 100644
--- a/lib/Runtime/ROCm/FlyRocmRuntimeWrappers.cpp
+++ b/lib/Runtime/ROCm/FlyRocmRuntimeWrappers.cpp
@@ -14,6 +14,7 @@
 
 #include <cassert>
 #include <cstdio>
+#include <cstdlib>
 #include <dlfcn.h>
 #include <vector>
 
@@ -41,8 +42,9 @@ extern "C" hipModule_t mgpuModuleLoad(void *data, size_t /*gpuBlobSize*/) {
 extern "C" hipModule_t mgpuModuleLoadJIT(void *data, int optLevel) {
   (void)data;
   (void)optLevel;
-  assert(false && "This function is not available in HIP.");
-  return nullptr;
+  fprintf(stderr, "[mgpuModuleLoadJIT] This function is not available in HIP.\n");
+  assert(false && "mgpuModuleLoadJIT is not available in HIP");
+  std::abort();
 }
 
 extern "C" void mgpuModuleUnload(hipModule_t module) {