• Hubert Lu's avatar
    Enable --transducer extension for ROCm (#88) · ae5ca671
    Hubert Lu authored
    * Enable --transducer extension for ROCm
    
    * Enable --transducer unit tests for ROCm
    
    * Skip some failing tests in test_transducer_joint.py
    
    * Skip test_transducer_joint_pack for transducer extension
    
    * Keep transducer extension CUDA-compatible
    ae5ca671
transducer_joint_kernel.cu 36.7 KB