-
Notifications
You must be signed in to change notification settings - Fork 1.6k
Open
Labels
Description
Which component requires the feature?
CuTe DSL
Feature Request
Is your feature request related to a problem? Please describe.
It would be very valuable to be able to integrate CuTeDSL kernels into existing codebases which have stronger type checking set up. Presently, when using mypy, typing must be largely ignored around CuTeDSL.
Describe the solution you'd like
Better type annotations + py.typed files in the CuTeDSL Python packages with a reasonable degree of type checking support via mypy.
Describe alternatives you've considered
Ignored types in kernels/launchers/code interfacing with CuTeDSL.
Additional context
N/A