Update on "[ET-VK][ez] Allow high dimensional tensors (for buffer storage)"

Differential Revision: [D80800083](https://our.internmc.facebook.com/intern/diff/D80800083)

[ghstack-poisoned]

Author: ssjia

backends/vulkan/runtime/api/containers/Tensor.cpp

@@ -847,7 +847,7 @@ const vkapi::BufferBindInfo vTensor::numel_ubo() {
 }
 
 const vkapi::BufferBindInfo vTensor::buffer_meta_ubo() {
-  size_t ubo_nbytes = std::min(sizeof(BufferMetadata), min_nbytes_per_ubo_);
+  size_t ubo_nbytes = std::max(sizeof(BufferMetadata), min_nbytes_per_ubo_);
   if (!buffer_meta_.buffer()) {
     BufferMetadata data(sizes_, dim_order_, strides_, numel_);
     buffer_meta_ = ParamsBuffer(storage_->context_, data);

backends/vulkan/runtime/api/containers/Tensor.h

@@ -548,7 +548,7 @@ class vTensor final {
     if (!uniforms_.buffer()) {
       uniforms_ = ParamsBuffer(storage_->context_, max_ubo_nbytes_, true);
     }
-    size_t ubo_nbytes = std::min(sizeof(T), min_nbytes_per_ubo_);
+    size_t ubo_nbytes = std::max(sizeof(T), min_nbytes_per_ubo_);
     if (*param_buffer_offset == kUniformOffsetUnset) {
       VK_CHECK_COND(
           (uniforms_size_ + ubo_nbytes) <= max_ubo_nbytes_,

backends/vulkan/runtime/graph/ops/glsl/indexing.glslh

@@ -187,7 +187,7 @@ void printTensorIndex(const TensorIndex tidx) {
 
 void printBufferMetadata(const BufferMetadata meta) {
     debugPrintfEXT(
-        "BufferMetadata: ndim=%u numel=%u\\n  sizes=[%u %u %u %u %u %u %u %u]\\n  dim_order=[%u %u %u %u %u %u %u %u]\\n strides=[%u %u %u %u %u %u %u %u]\\n",
+        "BufferMetadata: ndim=%u numel=%u\\n  sizes=[%u %u %u %u %u %u %u %u]\\n  dim_order=[%u %u %u %u %u %u %u %u]\\n  strides=[%u %u %u %u %u %u %u %u]\\n",
         meta.ndim_numel[0], meta.ndim_numel[1],
         meta.sizes[0][0], meta.sizes[0][1], meta.sizes[0][2], meta.sizes[0][3],
         meta.sizes[1][1], meta.sizes[1][1], meta.sizes[1][2], meta.sizes[1][3],