Skip to content

Commit

Permalink
Fix some logic in the PTX warning printed
Browse files Browse the repository at this point in the history
  • Loading branch information
Caspar van Leeuwen committed Feb 20, 2025
1 parent de6d49d commit 0e97868
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions easybuild/framework/easyblock.py
Original file line number Diff line number Diff line change
Expand Up @@ -3373,13 +3373,14 @@ def sanity_check_cuda(self, cuda_dirs=None, check_cuobjdump=True):
self.log.debug(msg)

# Check whether there is ptx code for the highest CC in cfg_ccs
highest_cc = sorted(cfg_ccs)[-1]
highest_cc = [sorted(cfg_ccs)[-1]]
missing_ptx_ccs = list(set(highest_cc) - set(derived_ptx_ccs))

if missing_ptx_ccs:
fail_msg = "Configured highest compute capability was '%s', "
fail_msg += "but no PTX code for this compute capability was found in '%s'"
self.log.warning(fail_msg, highest_cc, missing_ptx_ccs)
fail_msg += "but no PTX code for this compute capability was found in '%s' "
fail_msg += "PTX architectures supported in that file: %s"
self.log.warning(fail_msg, highest_cc[0], path, derived_ptx_ccs)
else:
msg = (f"Output of 'cuobjdump' checked for {path}; ptx code was present for (at least) the"
" highest CUDA compute capability in cuda_compute_capabilities")
Expand Down

0 comments on commit 0e97868

Please sign in to comment.