forked from ROCm/flash-attention
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathhipify_patch.patch
20 lines (20 loc) · 1.5 KB
/
hipify_patch.patch
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
@@ -816,10 +816,15 @@
return m.group(0)
# Hipify header file first if needed
if header_filepath not in HIPIFY_FINAL_RESULT:
- preprocess_file_and_save_result(output_directory,
- header_filepath,
- all_files, header_include_dirs, stats, hip_clang_launch,
- is_pytorch_extension, clean_ctx, show_progress)
+ #JCG added skip logic
+ if "composable_kernel" in header_filepath:
+ print("Force skipping hipification of CK file: " + header_filepath)
+ HIPIFY_FINAL_RESULT[header_filepath] = {"hipified_path":header_filepath}
+ else:
+ preprocess_file_and_save_result(output_directory,
+ header_filepath,
+ all_files, header_include_dirs, stats, hip_clang_launch,
+ is_pytorch_extension, clean_ctx, show_progress)
hipified_header_filepath = HIPIFY_FINAL_RESULT[header_filepath]["hipified_path"]
return templ.format(os.path.relpath(hipified_header_filepath if hipified_header_filepath is not None
else header_filepath, header_dir))