Skip to content

GitLab

Unverified Commit 907c3ff0 authored by botbw's avatar botbw Committed by GitHub
Browse files

[Example] Remove redundant param (#821)

parent d3e75b70
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/flash_attention/example_mha_fwd_bshd.py
View file @ 907c3ff0
...@@ -193,7 +193,7 @@ def main( ...@@ -193,7 +193,7 @@ def main(
print("Tile-lang: {:.2f} ms".format(latency)) print("Tile-lang: {:.2f} ms".format(latency))
print("Tile-lang: {:.2f} TFlops".format(total_flops / latency * 1e-9)) print("Tile-lang: {:.2f} TFlops".format(total_flops / latency * 1e-9))
else: else:
best_result = flashattn(batch, heads, seq_len, dim, is_causal, tune=tune) best_result = flashattn(batch, heads, seq_len, dim, is_causal)
best_latency = best_result.latency best_latency = best_result.latency
best_config = best_result.config best_config = best_result.config
ref_latency = best_result.ref_latency ref_latency = best_result.ref_latency
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment