code: def _check_max_workspace_size(
      buffer_pool_allocations, pool_info, size
  ):
      max_workspace_size = 0
      for (
          buffer_info,
          pool_allocation,
      ) in buffer_pool_allocations.items():
          if pool_allocation.pool_info == pool_info:
              size_candidate = (
                  pool_allocation.byte_offset + buffer_info.size_bytes
              )
              if size_candidate > max_workspace_size:
                  max_workspace_size = size_candidate
      assert max_workspace_size == size
  
imports: import pytest
  import tvm
  from tvm import tir, script
  from tvm.script import tir
  from tvm.tir import stmt_functor
  from tvm.tir.usmp import utils
  from tvm.target import Target
  from tvm import WorkspacePoolInfo, PoolInfoProperties
label_desc: Use image node_id 7 for calling a global function with example usage: _check_max_workspace_size(buffer_pool_allocations, pool_info, size) without return types