File tree Expand file tree Collapse file tree
llm-inference/gemma/src/jvmTest/kotlin/sk/ainet/models/gemma Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -56,7 +56,7 @@ class GemmaMlirDumpTest {
5656 val graph = (tape as DefaultExecutionTape ).toComputeGraph(synthesizeExternalInputs = true )
5757 val mlir = sk.ainet.compile.hlo.toStableHlo(graph, " gemma" ).content
5858
59- val out = File (System .getProperty(" gemmaMlirOut" ) ? : " /home/miso/projects/coral /build-mlir/gemma.mlir" )
59+ val out = File (System .getProperty(" gemmaMlirOut" ) ? : " build /build-mlir/gemma.mlir" )
6060 out .parentFile?.mkdirs()
6161 out .writeText(mlir)
6262 println (" WROTE_MLIR ${out .absolutePath} (${mlir.lines().size} lines)" )
Original file line number Diff line number Diff line change @@ -83,7 +83,7 @@ class RealGemmaMlirDumpTest {
8383 val unsupported = mlir.lines().count { it.contains(" Unsupported op" , ignoreCase = true ) }
8484 val arity = mlir.lines().count { it.contains(" arity" , ignoreCase = true ) }
8585 println (" MLIR lines=${mlir.lines().size} unsupported=$unsupported arity=$arity " )
86- val out = File (System .getProperty(" gemmaMlirOut" ) ? : " /home/miso/projects/coral /build-mlir/gemma-real.mlir" )
86+ val out = File (System .getProperty(" gemmaMlirOut" ) ? : " build /build-mlir/gemma-real.mlir" )
8787 out .parentFile?.mkdirs()
8888 out .writeText(mlir)
8989 println (" WROTE_MLIR ${out .absolutePath} " )
You can’t perform that action at this time.
0 commit comments