Repository: systemml Updated Branches: refs/heads/master f35cb6005 -> 7fb38b9b0
[SYSTEMML-445] Turn off batchnorm rewrite for mode="train" Project: http://git-wip-us.apache.org/repos/asf/systemml/repo Commit: http://git-wip-us.apache.org/repos/asf/systemml/commit/7fb38b9b Tree: http://git-wip-us.apache.org/repos/asf/systemml/tree/7fb38b9b Diff: http://git-wip-us.apache.org/repos/asf/systemml/diff/7fb38b9b Branch: refs/heads/master Commit: 7fb38b9b08155b2466e9711ec87a3a3a26256c32 Parents: f35cb60 Author: Niketan Pansare <[email protected]> Authored: Wed Aug 8 13:34:18 2018 -0700 Committer: Niketan Pansare <[email protected]> Committed: Wed Aug 8 13:34:51 2018 -0700 ---------------------------------------------------------------------- .../java/org/apache/sysml/hops/rewrite/RewriteGPUSpecificOps.java | 2 +- src/test/java/org/apache/sysml/test/gpu/BatchNormTest.java | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/systemml/blob/7fb38b9b/src/main/java/org/apache/sysml/hops/rewrite/RewriteGPUSpecificOps.java ---------------------------------------------------------------------- diff --git a/src/main/java/org/apache/sysml/hops/rewrite/RewriteGPUSpecificOps.java b/src/main/java/org/apache/sysml/hops/rewrite/RewriteGPUSpecificOps.java index f2a0588..1ee23f8 100644 --- a/src/main/java/org/apache/sysml/hops/rewrite/RewriteGPUSpecificOps.java +++ b/src/main/java/org/apache/sysml/hops/rewrite/RewriteGPUSpecificOps.java @@ -120,7 +120,7 @@ public class RewriteGPUSpecificOps extends HopRewriteRule { rule_GPUKernels(roots, hi, descendFirst); //see below if(roots != null) { - hi = batchNormTrain(roots, hop, hi, i); + //hi = batchNormTrain(roots, hop, hi, i); } hi = batchNormTest(hop, hi, i); hi = channelSums(hop, hi, i); http://git-wip-us.apache.org/repos/asf/systemml/blob/7fb38b9b/src/test/java/org/apache/sysml/test/gpu/BatchNormTest.java ---------------------------------------------------------------------- diff --git a/src/test/java/org/apache/sysml/test/gpu/BatchNormTest.java b/src/test/java/org/apache/sysml/test/gpu/BatchNormTest.java index b8bb9b6..d96feac 100644 --- a/src/test/java/org/apache/sysml/test/gpu/BatchNormTest.java +++ b/src/test/java/org/apache/sysml/test/gpu/BatchNormTest.java @@ -73,7 +73,7 @@ public class BatchNormTest extends GPUTests { } } else { - assertHeavyHitterPresent("gpu_batch_norm2d_train"); + //assertHeavyHitterPresent("gpu_batch_norm2d_train"); double [] threshold = new double[outputs.size()]; Arrays.fill(threshold, getTHRESHOLD()); // Handle loss of precision in CuDNN kernel
