diff --git a/userbenchmark/triton/ci.py b/userbenchmark/triton/ci.py index ec4664033..1b5659910 100644 --- a/userbenchmark/triton/ci.py +++ b/userbenchmark/triton/ci.py @@ -10,6 +10,7 @@ CI_TESTS = [ ["--op", "softmax", "--num-inputs", "10"], + ["--op", "sum"], ] def ci_result_to_userbenchmark_json(ci_metrics: List[BenchmarkOperatorResult]) -> Dict[str, Any]: