Merge pull request #4301 from gokul-uf/gpujoin_fix

theano/sandbox/gpuarray/basic_ops.py

@@ -997,6 +997,8 @@ class GpuJoin(HideC, Join):
     def perform(self, node, axis_and_tensors, out_, ctx):
         out, = out_
         axis = int(axis_and_tensors[0])
+        if axis < -axis_and_tensors[1].ndim:
+            raise IndexError
         if axis < 0:
             axis += axis_and_tensors[1].ndim
         tensors = axis_and_tensors[1:]

theano/tensor/tests/test_basic.py

@@ -3884,7 +3884,7 @@ class T_Join_and_Split(unittest.TestCase):
         got = f(-2)
         assert numpy.allclose(got, want)
 
-        self.assertRaises((IndexError, OverflowError), f, -3)
+        self.assertRaises(IndexError, f, -3)
 
     def test_join_matrixC_negative_axis(self):
         """constant join negative axis"""
@@ -3916,7 +3916,7 @@ class T_Join_and_Split(unittest.TestCase):
         got = f()
         assert numpy.allclose(got, want)
 
-        self.assertRaises((IndexError, OverflowError), join, -3, a, b)
+        self.assertRaises(IndexError, join, -3, a, b)
 
         utt.verify_grad(lambda a, b: join(-1, a, b), [v, 2 * v],
                         mode=self.mode)