Skip to content

Commit e23dca3

Browse files
committed
Add some @transient annotation
1 parent c1668e4 commit e23dca3

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

benchmarks/src/jmh/scala/com/thoughtworks/compute/benchmarks.scala

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ object benchmarks {
2626
@State(Scope.Benchmark)
2727
class Nd4jSigmoid extends SigmoidState {
2828

29+
@transient
2930
private lazy val input = Nd4j.randn(Array.fill(numberOfDimensions)(size))
3031
private def sigmoid(x: INDArray): INDArray = {
3132
val expX = Transforms.exp(x)
@@ -107,6 +108,7 @@ object benchmarks {
107108
@State(Scope.Benchmark)
108109
class Nd4jSum extends SumState {
109110

111+
@transient
110112
private lazy val input = Nd4j.randn(Array.fill(numberOfDimensions)(size))
111113

112114
@Benchmark
@@ -239,7 +241,11 @@ object benchmarks {
239241
@Threads(value = Threads.MAX)
240242
@State(Scope.Benchmark)
241243
class Nd4jConvolution extends ConvolutionState {
244+
245+
@transient
242246
private lazy val input = Nd4j.randn(Array(batchSize, depth, imageHeight, imageWidth))
247+
248+
@transient
243249
private lazy val layers = (for (i <- (0 until numberOfLayers).view) yield {
244250
(Nd4j.randn(Array(kernelHeight, kernelWidth, depth, depth)), Nd4j.randn(Array(depth)))
245251
}).toList

0 commit comments

Comments
 (0)