From 203f1166d32d8ab0e5487172cc1b799284368350 Mon Sep 17 00:00:00 2001
From: Junyuan Xie <eric.jy.xie@gmail.com>
Date: Sat, 5 Dec 2015 01:00:22 -0800
Subject: [PATCH] fix upsampling plan

---
 mshadow/extension/spatial_upsampling_nearest.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mshadow/extension/spatial_upsampling_nearest.h b/mshadow/extension/spatial_upsampling_nearest.h
index 3a731b6c899f..534fbdd9ebe0 100644
--- a/mshadow/extension/spatial_upsampling_nearest.h
+++ b/mshadow/extension/spatial_upsampling_nearest.h
@@ -50,7 +50,7 @@ struct Plan<UpSamplingNearestExp<SrcExp, DType, srcdim>, DType> {
     : src_(MakePlan(e.src_)),
       scale_(e.scale_),
       new_height_(e.shape_[srcdim - 2]),
-      src_height_(static_cast<index_t>(e.shape_[srcdim - 2] / 2)) {}
+      src_height_(static_cast<index_t>(e.shape_[srcdim - 2] / e.scale_)) {}
   MSHADOW_XINLINE DType Eval(index_t i, index_t j) const {
     const index_t x = j;
     const index_t y = i % new_height_;