Skip to content

Commit cd01bfa

Browse files
committed
Add cuda dependencies for benchmark
Note that nd4j is downgraded to 0.8.0 due to https://github.com/deeplearning4j/nd4j/issues/2767
1 parent c34f558 commit cd01bfa

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

benchmarks/build.sbt

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
enablePlugins(JmhPlugin)
22

3-
libraryDependencies += "org.nd4j" % "nd4j-api" % "0.9.1"
3+
libraryDependencies += "org.nd4j" % "nd4j-api" % "0.8.0"
44

5-
libraryDependencies += "org.nd4j" % "nd4j-native-platform" % "0.9.1"
5+
libraryDependencies += "org.nd4j" % "nd4j-cuda-8.0-platform" % "0.8.0"
6+
7+
libraryDependencies += "org.nd4j" % "nd4j-native-platform" % "0.8.0"
68

79
libraryDependencies += ("org.lwjgl" % "lwjgl" % "3.1.6").jar().classifier {
810
import scala.util.Properties._

0 commit comments

Comments
 (0)