code: def set_hbm_per_device(
      storage_constraint: Topology, hbm_per_device: int
  ) -> None:
      for device in storage_constraint.devices:
          device.storage.hbm = hbm_per_device
  
imports: import copy
  import heapq
  import logging
  from dataclasses import dataclass
  from enum import Enum
  from typing import cast, Dict, List, Optional
  from torchrec.distributed.planner.perf_models import NoopPerfModel
  from torchrec.distributed.planner.types import DeviceHardware, PartitionByType, Partitioner, Perf, PerfModel, PlannerError, PlannerErrorType, ShardingOption, Storage, Topology
  from torchrec.distributed.planner.utils import bytes_to_gb, reset_shard_rank
  from torchrec.distributed.types import ShardingType
label_desc: Use image node_id 4 for calling a global function with example usage: set_hbm_per_device(storage_constraint, hbm_per_device) without return types