Skip to content

Commit b23d937

Browse files
authored
Forward fix of forward only ops
Differential Revision: D84155460 Pull Request resolved: #533
1 parent bb14aa5 commit b23d937

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tritonbench/operators/decoding_attention/operator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -277,6 +277,7 @@ class Operator(BenchmarkOperator):
277277
DEFAULT_PRECISION = "bf16"
278278

279279
DEFAULT_METRICS = ["latency", "speedup"]
280+
FWD_ONLY = True
280281

281282
def __init__(
282283
self, tb_args: argparse.Namespace, extra_args: Optional[List[str]] = None

0 commit comments

Comments
 (0)