485e19e483e543fada5e64e50a9ab82cb6a91b24,src/graph_builder/backend/webgpu/meta_buffer_injector.py,MetaBufferInjector,generate_buffer,#MetaBufferInjector#,47

Before Change

        offset_map = {}
        offset = 0
        values = []
        for key, value in self.data.items():
            values += [value]
            offset_map[key] = offset
            offset += 1

After Change

            if isinstance(value, int):
                values += np.array([value], dtype=np.int32).tobytes()

            elif isinstance(value, float):
                values += np.array([value], dtype=np.float32).tobytes()

            elif isinstance(value, bytes):
                if len(value) % 4 != 0:
                    value += bytes(4 - (len(value) % 4))
                    
                values += value

            else:
                raise TypeError("MetaBufferInjector only supports int, float, or bytes contents.")

        self.offset_map = offset_map
        self.buffer = values

Instances