@@ -659,7 +659,7 @@ utils::GPUMemoryLayout vTensor::estimate_memory_layout() const {
659659
660660const  vkapi::BufferBindInfo vTensor::sizes_ubo () {
661661  if  (!uniforms_.buffer ()) {
662-     uniforms_ = ParamsBuffer (storage_.context_ , kMaxUniformBufferSize );
662+     uniforms_ = ParamsBuffer (storage_.context_ , kMaxUniformBufferSize ,  true );
663663  }
664664  if  (sizes_uniform_offset_ == kUniformOffsetUnset ) {
665665    VK_CHECK_COND (
@@ -674,7 +674,7 @@ const vkapi::BufferBindInfo vTensor::sizes_ubo() {
674674
675675const  vkapi::BufferBindInfo vTensor::strides_ubo () {
676676  if  (!uniforms_.buffer ()) {
677-     uniforms_ = ParamsBuffer (storage_.context_ , kMaxUniformBufferSize );
677+     uniforms_ = ParamsBuffer (storage_.context_ , kMaxUniformBufferSize ,  true );
678678  }
679679  if  (unsqueezed_strides_offset_ == kUniformOffsetUnset ) {
680680    VK_CHECK_COND (
@@ -691,7 +691,7 @@ const vkapi::BufferBindInfo vTensor::strides_ubo() {
691691
692692const  vkapi::BufferBindInfo vTensor::logical_limits_ubo () {
693693  if  (!uniforms_.buffer ()) {
694-     uniforms_ = ParamsBuffer (storage_.context_ , kMaxUniformBufferSize );
694+     uniforms_ = ParamsBuffer (storage_.context_ , kMaxUniformBufferSize ,  true );
695695  }
696696  if  (logical_limits_uniform_offset_ == kUniformOffsetUnset ) {
697697    VK_CHECK_COND (
@@ -707,7 +707,7 @@ const vkapi::BufferBindInfo vTensor::logical_limits_ubo() {
707707
708708const  vkapi::BufferBindInfo vTensor::numel_ubo () {
709709  if  (!uniforms_.buffer ()) {
710-     uniforms_ = ParamsBuffer (storage_.context_ , kMaxUniformBufferSize );
710+     uniforms_ = ParamsBuffer (storage_.context_ , kMaxUniformBufferSize ,  true );
711711  }
712712  if  (numel_uniform_offset_ == kUniformOffsetUnset ) {
713713    VK_CHECK_COND (
0 commit comments