@@ -2153,19 +2153,30 @@ def sl_polybench_runner(polybench_run: mx_polybench.PolybenchRunFunction, tags:
21532153 mx_polybench .register_polybench_benchmark_suite (mx_suite = _suite , name = "sl" , languages = ["sl" ], benchmark_distribution = "SL_BENCHMARKS" , benchmark_file_filter = ".*sl" , runner = sl_polybench_runner , tags = {"gate" , "benchmark" })
21542154
21552155 # NFI benchmarks use Graal.js to test upcalls/downcalls, so only register them if Graal.js is available.
2156- # TODO (GR-64439): re-enable PMH after graaljs polybench migration
2157- # if mx.dependency("PMH", fatalIfMissing=False) and mx.suite("graal-js", fatalIfMissing=False):
2158- # mx_polybench.register_polybench_language(mx_suite=_suite, language="pmh", distributions=["PMH"], native_distributions=["PMH_BENCHMARK_NATIVE"])
2159-
2160-
2161- # def nfi_polybench_runner(polybench_run: mx_polybench.PolybenchRunFunction, tags) -> None:
2162- # if "gate" in tags:
2163- # polybench_run(["--jvm", "nfi/*.pmh", "--experimental-options", "--engine.Compilation=false", "-w", "1", "-i", "1"])
2164- # polybench_run(["--native", "nfi/*.pmh", "--experimental-options", "--engine.Compilation=false", "-w", "1", "-i", "1"])
2165- # polybench_run(["--jvm", "nfi/panama/*.pmh", "--experimental-options", "--engine.Compilation=false", "-w", "1", "-i", "1"])
2166- # if "benchmark" in tags:
2167- # polybench_run(["--jvm", "nfi/*.pmh"])
2168- # polybench_run(["--native", "nfi/*.pmh"])
2169- # polybench_run(["--jvm", "nfi/panama/*.pmh"])
2170-
2171- # mx_polybench.register_polybench_benchmark_suite(mx_suite=_suite, name="nfi", languages=["pmh", "js"], benchmark_distribution="NFI_POLYBENCH_BENCHMARKS", benchmark_file_filter=".*pmh", runner=nfi_polybench_runner, tags={"gate", "benchmark"})
2156+ if mx .dependency ("PMH" , fatalIfMissing = False ) and mx .suite ("graal-js" , fatalIfMissing = False ):
2157+ mx_polybench .register_polybench_language (mx_suite = _suite , language = "pmh" , distributions = ["PMH" ], native_distributions = ["PMH_BENCHMARK_NATIVE" ])
2158+
2159+ def nfi_polybench_runner (polybench_run : mx_polybench .PolybenchRunFunction , tags ) -> None :
2160+ if "gate" in tags :
2161+ polybench_run (["--jvm" , "nfi/*.pmh" , "--experimental-options" , "--engine.Compilation=false" , "-w" , "1" , "-i" , "1" ])
2162+ polybench_run (["--jvm" , "nfi/panama/*.pmh" , "--experimental-options" , "--engine.Compilation=false" , "-w" , "1" , "-i" , "1" ])
2163+ polybench_run (["--native" , "nfi/*.pmh" , "--experimental-options" , "--engine.Compilation=false" , "-w" , "1" , "-i" , "1" ])
2164+ if "benchmark" in tags :
2165+ polybench_run (["--jvm" , "nfi/*.pmh" , "--experimental-options" , "--engine.Compilation=false" ])
2166+ polybench_run (["--jvm" , "nfi/panama/*.pmh" , "--experimental-options" , "--engine.Compilation=false" ])
2167+ polybench_run (["--native" , "nfi/*.pmh" , "--experimental-options" , "--engine.Compilation=false" ])
2168+ polybench_run (["--jvm" , "nfi/*.pmh" ])
2169+ polybench_run (["--jvm" , "nfi/panama/*.pmh" ])
2170+ polybench_run (["--native" , "nfi/*.pmh" ])
2171+ polybench_run (["--jvm" , "nfi/*.pmh" , "--metric=metaspace-memory" ])
2172+ polybench_run (["--jvm" , "nfi/panama/*.pmh" , "--metric=metaspace-memory" ])
2173+ polybench_run (["--jvm" , "nfi/*.pmh" , "--metric=application-memory" ])
2174+ polybench_run (["--jvm" , "nfi/panama/*.pmh" , "--metric=application-memory" ])
2175+ polybench_run (["--jvm" , "nfi/*.pmh" , "--metric=allocated-bytes" , "-w" , "40" , "-i" , "10" , "--experimental-options" , "--engine.Compilation=false" ])
2176+ polybench_run (["--jvm" , "nfi/panama/*.pmh" , "--metric=allocated-bytes" , "-w" , "40" , "-i" , "10" , "--experimental-options" , "--engine.Compilation=false" ])
2177+ polybench_run (["--native" , "nfi/*.pmh" , "--metric=allocated-bytes" , "-w" , "40" , "-i" , "10" , "--experimental-options" , "--engine.Compilation=false" ])
2178+ polybench_run (["--jvm" , "nfi/*.pmh" , "--metric=allocated-bytes" , "-w" , "40" , "-i" , "10" ])
2179+ polybench_run (["--jvm" , "nfi/panama/*.pmh" , "--metric=allocated-bytes" , "-w" , "40" , "-i" , "10" ])
2180+ polybench_run (["--native" , "nfi/*.pmh" , "--metric=allocated-bytes" , "-w" , "40" , "-i" , "10" ])
2181+
2182+ mx_polybench .register_polybench_benchmark_suite (mx_suite = _suite , name = "nfi" , languages = ["pmh" , "js" ], benchmark_distribution = "NFI_POLYBENCH_BENCHMARKS" , benchmark_file_filter = ".*pmh" , runner = nfi_polybench_runner , tags = {"gate" , "benchmark" })
0 commit comments