Simple Microbenchmarks for ND4J utilizing the JMH
Switch branches/tags
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.mvn/wrapper
src/main/java/com/example
.gitignore
README.md
mvnw
mvnw.cmd
pom.xml

README.md

benchmarking_nd4j

Simple Microbenchmarks for ND4J utilizing the JMH

Building

mvn clean package

Running

java -jar target/benchmarks.jar

For a quick run, e.g. just the same Matrix Size progression as used in the Neanderthal benchmarks, run like this:

java -jar target/benchmarks.jar -f2 -i10 -wi 2 Neanderthal

Choosing a BLAS Library

Since ND4J supports multiple blas libraries, you have to specify which one you actually want to use. For my own benchmarks I've been using MKL.

-Dorg.bytedeco.javacpp.openblas.load=mkl_rt

For more information see https://github.com/bytedeco/javacpp-presets/tree/master/openblas