You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
Add subscript-style type annotations for Warp types #1216
dtype-first parameter order â aligns with NumPy ndarray[Dtype, Shape], C++ std::array<T, N>, Rust [T; N]
Instance-based returns â subscript results are the same objects as constructor/factory results, preserving compatibility with codegen's 68+ isinstance checks
Literal[] required for type checkers â bare integers work at runtime only
Tile shape is optional â wp.tile[dtype] for annotations, wp.tile[dtype, M, N] for forward compatibility
Activity
shi-eric commentedon Feb 15, 2026
Added in b7b75e2
Add subscript-style type hints for array and tile types (NVIDIAGH-1216)
Fix outdated Vector/Matrix docstrings to use dtype-first order (NVIDIâĶ
Improve type-related error messages (NVIDIAGH-1216)
Fix JAX FFI to support subscript-style array type hints (NVIDIAGH-1216)
Fix Tape to support subscript-style array type hints in structs (NVIDâĶ
Merge branch 'ershi/typing-improvements' into 'main'
Use subscript-style type annotations in examples
Add subscript-style type hints for array and tile types (NVIDIAGH-1216)
Fix outdated Vector/Matrix docstrings to use dtype-first order (NVIDIâĶ
Improve type-related error messages (NVIDIAGH-1216)
Fix JAX FFI to support subscript-style array type hints (NVIDIAGH-1216)
Fix Tape to support subscript-style array type hints in structs (NVIDâĶ