try MUSA fix

This commit is contained in:
Johannes Gäßler 2025-01-05 21:41:53 +01:00
parent fa77f7a41e
commit db8d6b71ec

View file

@ -133,3 +133,5 @@
#define cudaKernelNodeParams musaKernelNodeParams #define cudaKernelNodeParams musaKernelNodeParams
#define cudaStreamCaptureModeRelaxed musaStreamCaptureModeRelaxed #define cudaStreamCaptureModeRelaxed musaStreamCaptureModeRelaxed
#define cudaStreamEndCapture musaStreamEndCapture #define cudaStreamEndCapture musaStreamEndCapture
typedef mt_bfloat16 nv_bfloat16;