Skip to content

Commit b701de9

Browse files
Applied remarks
1 parent 1d0ecee commit b701de9

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

clang/lib/Driver/ToolChains/Flang.cpp

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -344,8 +344,18 @@ void Flang::AddAMDGPUTargetArgs(const ArgList &Args,
344344
}
345345

346346
// Check ROCm path if specified
347+
if (!Args.getLastArg(options::OPT_rocm_path_EQ))
348+
return;
349+
350+
// If ROCm path is specified, check correctness of the path
347351
const ToolChain &TC = getToolChain();
348-
TC.getDeviceLibs(Args);
352+
std::string HIPVersion;
353+
llvm::raw_string_ostream HIPInfo(HIPVersion);
354+
TC.printVerboseInfo(HIPInfo);
355+
llvm::StringRef HIPInfoStrRef(HIPInfo.str());
356+
if (!HIPInfoStrRef.contains("Found HIP installation") &&
357+
!Args.getLastArg(options::OPT_nogpulib))
358+
TC.getDriver().Diag(diag::err_drv_no_rocm_device_lib) << 0;
349359
}
350360

351361
void Flang::addTargetOptions(const ArgList &Args,

flang/test/Driver/omp-driver-offload.f90

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -202,10 +202,9 @@
202202
! RUN: | FileCheck --check-prefix=NO-LIBC-GPU-AMDGPU %s
203203
! NO-LIBC-GPU-AMDGPU-NOT: "-lcgpu-amdgpu"
204204

205-
! RUN: rm -rf %t/Inputs
206205
! RUN: not %flang -### -v --target=x86_64-unknown-linux-gnu -fopenmp \
207206
! RUN: --offload-arch=gfx900 \
208-
! RUN: --rocm-path=%t/Inputs/rocm %s 2>&1 \
207+
! RUN: --rocm-path=%S/Inputs/rocm %s 2>&1 \
209208
! RUN: | FileCheck --check-prefix=ROCM-PATH-NOT-FOUND %s
210209
! ROCM-PATH-NOT-FOUND: error: cannot find ROCm device library;
211210

@@ -214,7 +213,7 @@
214213
! RUN: cp -r %S/../../../clang/test/Driver/Inputs/rocm %t/Inputs
215214
! RUN: mkdir -p %t/Inputs/rocm/share/hip
216215
! RUN: mkdir -p %t/Inputs/rocm/hip
217-
! RUN: mv %t/Inputs/rocm/bin/.hipVersion %t/Inputs/rocm/share/hip/version
216+
! RUN: mv %t/Inputs/rocm/bin/.hipVersion %t/Inputs/rocm/share/hip/version
218217

219218
! RUN: %flang -### -v --target=x86_64-unknown-linux-gnu -fopenmp \
220219
! RUN: --offload-arch=gfx900 \

0 commit comments

Comments
 (0)