Update to new version of base image
This commit is contained in:
@@ -808,8 +808,6 @@ if triton.__version__ >= "2.1.0":
|
||||
)
|
||||
return
|
||||
|
||||
import time
|
||||
ts_beg = time.time()
|
||||
_fwd_kernel[grid](
|
||||
q,
|
||||
k,
|
||||
@@ -860,6 +858,4 @@ if triton.__version__ >= "2.1.0":
|
||||
num_warps=NUM_WARPS,
|
||||
num_stages=1,
|
||||
)
|
||||
elapsed = time.time() - ts_beg
|
||||
#print(f'{elapsed}: {BLOCK=}, {Lk=}, {Lk_padded=}, {BLOCK=}, {sliding_window=}, {NUM_WARPS=}')
|
||||
return
|
||||
|
||||
Reference in New Issue
Block a user