fix: Resolve cuda 710 error when compiling BERT models

- Issue arising when compiling BERT models with 3+ inputs
- Added temporary fix by decreasing the range of allowed values to the
random number generator for creating input tensors to 0-1, instead of 0-4
- More robust fix to follow

Author: gs-olive

core/partitioning/shape_analysis.cpp

@@ -13,15 +13,15 @@ at::Tensor generateSingleInput(ir::Input& input, c10::optional<at::ScalarType>&
   auto cur_shape = input.input_shape;
   std::vector<int64_t> shape;
   shape.insert(shape.begin(), std::begin(cur_shape.d), std::begin(cur_shape.d) + cur_shape.nbDims);
-  // auto type_opt = types[input.first][i];
+
   auto type = at::kFloat;
   if (type_opt) {
     type = type_opt.value();
   } else {
     LOG_WARNING("Input type for doing shape analysis could not be determined, defaulting to F32");
   }
-  auto in = at::randint(5, shape, {at::kCUDA}).to(type);
-  // ivalue_map[input.first] = in.clone();
+  auto in = at::randint(2, shape, {at::kCUDA}).to(type);
+
   return in;
 }