Improve prodall CUDA test

Author: apaszke

test/test_cuda.py

@@ -59,6 +59,13 @@ def small_2d_scaled(t, scale=10):
  return make_tensor(t, S, S).mul(scale)
 
 
+def small_2d_oneish(t):
+ if is_floating(t):
+ return make_tensor(t, S, S).clamp(min=0.99, max=1.01)
+ else:
+ return t(S, S).fill_(1)
+
+
 def small_3d(t):
  return make_tensor(t, S, S, S)
 
@@ -206,7 +213,7 @@ def tmp(t):
  ('norm', small_3d, lambda t: [3, 0], '3_norm_dim'),
  ('ones', small_3d, lambda t: [1, 2, 3, 4, 5],),
  ('permute', new_t(1, 2, 3, 4), lambda t: [2, 1, 3, 0],),
- ('prod', small_3d, lambda t: [],),
+ ('prod', small_2d_oneish, lambda t: [],),
  ('prod', small_3d, lambda t: [1], 'dim'),
  ('sum', small_2d, lambda t: [],),
  ('sum', small_3d, lambda t: [1], 'dim'),