You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@systemml.apache.org by ni...@apache.org on 2018/08/08 20:36:28 UTC
systemml git commit: [SYSTEMML-445] Turn off batchnorm rewrite for
mode="train"
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 <np...@us.ibm.com>
Authored: Wed Aug 8 13:34:18 2018 -0700
Committer: Niketan Pansare <np...@us.ibm.com>
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