Skip to content

Commit

Permalink
Modify some of the logs
Browse files Browse the repository at this point in the history
Signed-off-by: erman-gurses <erman@nod-labs.com>
  • Loading branch information
erman-gurses committed Mar 5, 2025
1 parent f0903bc commit 1a357ba
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions sharktank/tests/models/llama/benchmark_amdgpu_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -309,7 +309,7 @@ def testBenchmark8B_fp8_Non_Decomposed(self):
)


logger.info("BenchmarkLlama3_1_70B...")
logger.info("Starting BenchmarkLlama3_1_70B...")


@is_mi300x
Expand Down Expand Up @@ -606,7 +606,7 @@ def testBenchmark70B_fp8_TP8_Decomposed(self):
cwd=self.repo_root,
)

logger.info("Testing testBenchmark70B_fp8_TP8_Non_Decomposed...")
logger.info("Testing Benchmark70B_fp8_TP8_Non_Decomposed...")

@pytest.mark.xfail(
reason="70b fp8 irpa does not exist", strict=True, raises=ExportMlirException
Expand Down Expand Up @@ -660,7 +660,7 @@ def testBenchmark70B_fp8_TP8_Non_Decomposed(self):
)


logger.info("Benchmark Llama3_1_405B...")
logger.info("Starting Benchmark Llama3_1_405B...")


@is_mi300x
Expand Down

0 comments on commit 1a357ba

Please sign in to comment.