-
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathwhisper.patch
More file actions
32 lines (29 loc) · 894 Bytes
/
Copy pathwhisper.patch
File metadata and controls
32 lines (29 loc) · 894 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
--- whisper/triton_ops.py.orig 2025-05-22 14:08:23.116969492 -0500
+++ whisper/triton_ops.py 2025-05-22 14:09:14.882997993 -0500
@@ -60,7 +60,8 @@
tl.store(y_ptr + offsets, MIDDLE_ROW_HERE, mask=mask) # noqa: F821
kernel = triton.JITFunction(kernel.fn)
- kernel.src = kernel.src.replace(
+ src = kernel.src
+ src = src.replace(
" LOAD_ALL_ROWS_HERE",
"\n".join(
[
@@ -69,7 +70,7 @@
]
),
)
- kernel.src = kernel.src.replace(
+ src = src.replace(
" BUBBLESORT_HERE",
"\n\n".join(
[
@@ -90,7 +91,9 @@
]
),
)
- kernel.src = kernel.src.replace("MIDDLE_ROW_HERE", f"row{filter_width // 2}")
+ src = src.replace("MIDDLE_ROW_HERE", f"row{filter_width // 2}")
+ kernel._unsafe_update_src(src)
+ kernel.hash = None
return kernel