TransformerEngine
TransformerEngine copied to clipboard
Replace TE check_support with FE check_support
trafficstars
Is your feature request related to a problem? Please describe. The logic around cuDNN's support matrix for SDPA is getting long and hard to maintain.
Describe the solution you'd like We need to move to cudnn-frontend's logic for condition checks.
Describe alternatives you've considered N/A
Additional context N/A