Skip to content

Commit 15931fd

Browse files
ax3lgithub-actions[bot]
authored andcommitted
Update Stub Files
1 parent db7c1de commit 15931fd

File tree

3 files changed

+69
-3
lines changed

3 files changed

+69
-3
lines changed

src/amrex/space1d/amrex_1d_pybind/__init__.pyi

Lines changed: 23 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -307,7 +307,29 @@ class AmrMesh:
307307
def verbose(self) -> int: ...
308308

309309
class Arena:
310-
pass
310+
@staticmethod
311+
def finalize() -> None: ...
312+
@staticmethod
313+
def initialize() -> None: ...
314+
@staticmethod
315+
def print_usage() -> None: ...
316+
@staticmethod
317+
def print_usage_to_files(filename: str, message: str) -> None: ...
318+
def has_free_device_memory(self, sz: int) -> bool:
319+
"""
320+
Does the device have enough free memory for allocating this much memory? For CPU builds, this always return true.
321+
"""
322+
323+
@property
324+
def is_device(self) -> bool: ...
325+
@property
326+
def is_device_accessible(self) -> bool: ...
327+
@property
328+
def is_host_accessible(self) -> bool: ...
329+
@property
330+
def is_managed(self) -> bool: ...
331+
@property
332+
def is_pinned(self) -> bool: ...
311333

312334
class Array4_cdouble:
313335
@typing.overload

src/amrex/space2d/amrex_2d_pybind/__init__.pyi

Lines changed: 23 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,29 @@ class AmrMesh:
330330
def verbose(self) -> int: ...
331331

332332
class Arena:
333-
pass
333+
@staticmethod
334+
def finalize() -> None: ...
335+
@staticmethod
336+
def initialize() -> None: ...
337+
@staticmethod
338+
def print_usage() -> None: ...
339+
@staticmethod
340+
def print_usage_to_files(filename: str, message: str) -> None: ...
341+
def has_free_device_memory(self, sz: int) -> bool:
342+
"""
343+
Does the device have enough free memory for allocating this much memory? For CPU builds, this always return true.
344+
"""
345+
346+
@property
347+
def is_device(self) -> bool: ...
348+
@property
349+
def is_device_accessible(self) -> bool: ...
350+
@property
351+
def is_host_accessible(self) -> bool: ...
352+
@property
353+
def is_managed(self) -> bool: ...
354+
@property
355+
def is_pinned(self) -> bool: ...
334356

335357
class Array4_cdouble:
336358
@typing.overload

src/amrex/space3d/amrex_3d_pybind/__init__.pyi

Lines changed: 23 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -307,7 +307,29 @@ class AmrMesh:
307307
def verbose(self) -> int: ...
308308

309309
class Arena:
310-
pass
310+
@staticmethod
311+
def finalize() -> None: ...
312+
@staticmethod
313+
def initialize() -> None: ...
314+
@staticmethod
315+
def print_usage() -> None: ...
316+
@staticmethod
317+
def print_usage_to_files(filename: str, message: str) -> None: ...
318+
def has_free_device_memory(self, sz: int) -> bool:
319+
"""
320+
Does the device have enough free memory for allocating this much memory? For CPU builds, this always return true.
321+
"""
322+
323+
@property
324+
def is_device(self) -> bool: ...
325+
@property
326+
def is_device_accessible(self) -> bool: ...
327+
@property
328+
def is_host_accessible(self) -> bool: ...
329+
@property
330+
def is_managed(self) -> bool: ...
331+
@property
332+
def is_pinned(self) -> bool: ...
311333

312334
class Array4_cdouble:
313335
@typing.overload

0 commit comments

Comments
 (0)