Skip to content

Keep CUDA-Samples hook for version 12.9 #270

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ocaisa wants to merge 1 commit into
base: main
Choose a base branch
from
+9 −1
Open

Keep CUDA-Samples hook for version 12.9 #270

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 9 additions & 1 deletion eb_hooks.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1142,17 +1142,25 @@ def pre_configure_hook_BLIS(self, *args, **kwargs):
def pre_configure_hook_CUDA_Samples_test_remove(self, *args, **kwargs):
"""skip immaTensorCoreGemm in CUDA-Samples for compute capability 7.0."""
if self.name == 'CUDA-Samples':
if self.version in ['12.1']:
if self.version in ['12.1', '12.9']:
# Get compute capability from build option
cuda_caps = build_option('cuda_compute_capabilities')
# Check if compute capability 7.0 is in the list
if cuda_caps and '7.0' in cuda_caps:
print_msg("Applying hook for CUDA-Samples %s with compute capability 7.0", self.version)
# local_filters is set by the easyblock, remove path to the Makefile instead
makefile_path = os.path.join(self.start_dir, 'Samples/3_CUDA_Features/immaTensorCoreGemm/Makefile')
# later versions use CMake rather than Makefiles
cmakefile_path = os.path.join(self.start_dir, 'Samples/3_CUDA_Features/CMakeLists.txt')
if os.path.exists(makefile_path):
remove_file(makefile_path)
print_msg("Removed Makefile at %s to skip immaTensorCoreGemm build", makefile_path)
elif os.path.exists(cmakefile_path):
self.cfg.update(
'preconfigopts',
# Comment out the line that contains immaTensorCoreGemm
"sed -i '17{/immaTensorCoreGemm/s/^/#/}' %s && " % cmakefile_path,
)
else:
print_msg("Makefile not found at %s", makefile_path)
else:
Expand Down
Loading