File size: 260 Bytes
07fb183
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
"""Canonical CUDA memory ownership contracts."""

from __future__ import annotations

from dataclasses import dataclass


@dataclass(frozen=True, slots=True)
class DeviceAllocation:
    """One owned CUDA device allocation."""

    ptr: int
    size_bytes: int