"""
@generated by mypy-protobuf.  Do not edit manually!
isort:skip_file
"""

import builtins
import collections.abc
import typing

import google.protobuf.descriptor
import google.protobuf.internal.containers
import google.protobuf.message

DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

@typing.final
class MemAllocatorStats(google.protobuf.message.Message):
    """Some of the data from AllocatorStats"""

    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    NUM_ALLOCS_FIELD_NUMBER: builtins.int
    BYTES_IN_USE_FIELD_NUMBER: builtins.int
    PEAK_BYTES_IN_USE_FIELD_NUMBER: builtins.int
    LARGEST_ALLOC_SIZE_FIELD_NUMBER: builtins.int
    FRAGMENTATION_METRIC_FIELD_NUMBER: builtins.int
    num_allocs: builtins.int
    bytes_in_use: builtins.int
    peak_bytes_in_use: builtins.int
    largest_alloc_size: builtins.int
    fragmentation_metric: builtins.float
    def __init__(
        self,
        *,
        num_allocs: builtins.int | None = ...,
        bytes_in_use: builtins.int | None = ...,
        peak_bytes_in_use: builtins.int | None = ...,
        largest_alloc_size: builtins.int | None = ...,
        fragmentation_metric: builtins.float | None = ...,
    ) -> None: ...
    def ClearField(self, field_name: typing.Literal["bytes_in_use", b"bytes_in_use", "fragmentation_metric", b"fragmentation_metric", "largest_alloc_size", b"largest_alloc_size", "num_allocs", b"num_allocs", "peak_bytes_in_use", b"peak_bytes_in_use"]) -> None: ...

global___MemAllocatorStats = MemAllocatorStats

@typing.final
class MemChunk(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    ADDRESS_FIELD_NUMBER: builtins.int
    SIZE_FIELD_NUMBER: builtins.int
    REQUESTED_SIZE_FIELD_NUMBER: builtins.int
    BIN_FIELD_NUMBER: builtins.int
    OP_NAME_FIELD_NUMBER: builtins.int
    FREED_AT_COUNT_FIELD_NUMBER: builtins.int
    ACTION_COUNT_FIELD_NUMBER: builtins.int
    IN_USE_FIELD_NUMBER: builtins.int
    STEP_ID_FIELD_NUMBER: builtins.int
    address: builtins.int
    size: builtins.int
    requested_size: builtins.int
    bin: builtins.int
    op_name: builtins.str
    freed_at_count: builtins.int
    action_count: builtins.int
    in_use: builtins.bool
    step_id: builtins.int
    def __init__(
        self,
        *,
        address: builtins.int | None = ...,
        size: builtins.int | None = ...,
        requested_size: builtins.int | None = ...,
        bin: builtins.int | None = ...,
        op_name: builtins.str | None = ...,
        freed_at_count: builtins.int | None = ...,
        action_count: builtins.int | None = ...,
        in_use: builtins.bool | None = ...,
        step_id: builtins.int | None = ...,
    ) -> None: ...
    def ClearField(self, field_name: typing.Literal["action_count", b"action_count", "address", b"address", "bin", b"bin", "freed_at_count", b"freed_at_count", "in_use", b"in_use", "op_name", b"op_name", "requested_size", b"requested_size", "size", b"size", "step_id", b"step_id"]) -> None: ...

global___MemChunk = MemChunk

@typing.final
class BinSummary(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    BIN_FIELD_NUMBER: builtins.int
    TOTAL_BYTES_IN_USE_FIELD_NUMBER: builtins.int
    TOTAL_BYTES_IN_BIN_FIELD_NUMBER: builtins.int
    TOTAL_CHUNKS_IN_USE_FIELD_NUMBER: builtins.int
    TOTAL_CHUNKS_IN_BIN_FIELD_NUMBER: builtins.int
    bin: builtins.int
    total_bytes_in_use: builtins.int
    total_bytes_in_bin: builtins.int
    total_chunks_in_use: builtins.int
    total_chunks_in_bin: builtins.int
    def __init__(
        self,
        *,
        bin: builtins.int | None = ...,
        total_bytes_in_use: builtins.int | None = ...,
        total_bytes_in_bin: builtins.int | None = ...,
        total_chunks_in_use: builtins.int | None = ...,
        total_chunks_in_bin: builtins.int | None = ...,
    ) -> None: ...
    def ClearField(self, field_name: typing.Literal["bin", b"bin", "total_bytes_in_bin", b"total_bytes_in_bin", "total_bytes_in_use", b"total_bytes_in_use", "total_chunks_in_bin", b"total_chunks_in_bin", "total_chunks_in_use", b"total_chunks_in_use"]) -> None: ...

global___BinSummary = BinSummary

@typing.final
class SnapShot(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    ACTION_COUNT_FIELD_NUMBER: builtins.int
    SIZE_FIELD_NUMBER: builtins.int
    action_count: builtins.int
    size: builtins.int
    def __init__(
        self,
        *,
        action_count: builtins.int | None = ...,
        size: builtins.int | None = ...,
    ) -> None: ...
    def ClearField(self, field_name: typing.Literal["action_count", b"action_count", "size", b"size"]) -> None: ...

global___SnapShot = SnapShot

@typing.final
class MemoryDump(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    ALLOCATOR_NAME_FIELD_NUMBER: builtins.int
    BIN_SUMMARY_FIELD_NUMBER: builtins.int
    CHUNK_FIELD_NUMBER: builtins.int
    SNAP_SHOT_FIELD_NUMBER: builtins.int
    STATS_FIELD_NUMBER: builtins.int
    allocator_name: builtins.str
    @property
    def bin_summary(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BinSummary]: ...
    @property
    def chunk(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___MemChunk]: ...
    @property
    def snap_shot(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___SnapShot]: ...
    @property
    def stats(self) -> global___MemAllocatorStats: ...
    def __init__(
        self,
        *,
        allocator_name: builtins.str | None = ...,
        bin_summary: collections.abc.Iterable[global___BinSummary] | None = ...,
        chunk: collections.abc.Iterable[global___MemChunk] | None = ...,
        snap_shot: collections.abc.Iterable[global___SnapShot] | None = ...,
        stats: global___MemAllocatorStats | None = ...,
    ) -> None: ...
    def HasField(self, field_name: typing.Literal["stats", b"stats"]) -> builtins.bool: ...
    def ClearField(self, field_name: typing.Literal["allocator_name", b"allocator_name", "bin_summary", b"bin_summary", "chunk", b"chunk", "snap_shot", b"snap_shot", "stats", b"stats"]) -> None: ...

global___MemoryDump = MemoryDump
