Skip to content

Commit ab1b538

Browse files
[lang] Add example array types built on static eval
Signed-off-by: Asher Mancinelli <amancinelli@nvidia.com>
1 parent 2bbcaba commit ab1b538

3 files changed

Lines changed: 736 additions & 0 deletions

File tree

experimental/cuda-lang/src/cuda/lang/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -266,6 +266,8 @@
266266
clusterlaunchcontrol_token,
267267
)
268268

269+
from . import _debug
270+
269271
__all__ = (
270272
"dtype_of",
271273
"pointer_dtype",
@@ -480,4 +482,5 @@
480482
"MatrixLoadShape",
481483
"MatrixLoadSourceFormat",
482484
"MatrixStoreShape",
485+
"_debug",
483486
)
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
# SPDX-FileCopyrightText: Copyright (c) <2025> NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
#
3+
# SPDX-License-Identifier: Apache-2.0
4+
5+
from ._execution import static_def
6+
7+
8+
@static_def
9+
def static_inspect(*args, **kwargs):
10+
breakpoint()
11+
12+
13+
@static_def
14+
def static_print(*args, **kwargs):
15+
print(*args, **kwargs)

0 commit comments

Comments
 (0)