Skip to content

Commit 0e213df

Browse files
author
ssjia
committed
Update on "[ET-VK][ez] Consolidate tensor metadata calculation + buffer binding code"
Differential Revision: [D80800085](https://our.internmc.facebook.com/intern/diff/D80800085) [ghstack-poisoned]
2 parents 6b34c30 + d0c5ba6 commit 0e213df

File tree

1 file changed

+1
-1
lines changed
  • backends/vulkan/runtime/api/containers

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -521,7 +521,7 @@ class vTensor final {
521521
if (!uniforms_.buffer()) {
522522
uniforms_ = ParamsBuffer(storage_->context_, max_ubo_nbytes_, true);
523523
}
524-
size_t ubo_nbytes = std::min(sizeof(T), min_nbytes_per_ubo_);
524+
size_t ubo_nbytes = std::max(sizeof(T), min_nbytes_per_ubo_);
525525
if (*param_buffer_offset == kUniformOffsetUnset) {
526526
VK_CHECK_COND(
527527
(uniforms_size_ + ubo_nbytes) <= max_ubo_nbytes_,

0 commit comments

Comments
 (0)