Enable --transducer extension for ROCm

hubertlu-tw commented 2 years ago

There are five transducer extension unit tests skipped on ROCm. Please refer to https://github.com/ROCmSoftwarePlatform/apex/issues/89 for the details.

test_transducer_joint_pack_relu_dropout (test_transducer_joint.TransducerJointTest)
test_transducer_joint_relu_dropout (test_transducer_joint.TransducerJointTest)
test_transducer_joint_vec_pack_relu_dropout (test_transducer_joint.TransducerJointTest)
test_transducer_joint_vec_relu_dropout (test_transducer_joint.TransducerJointTest)
test_transducer_joint_pack_relu (test_transducer_joint.TransducerJointTest)

hubertlu-tw commented 2 years ago

jenkins: retest this please

hubertlu-tw commented 2 years ago

jenkins: retest this please

hubertlu-tw commented 2 years ago

jenkins: retest this please

jithunnair-amd commented 2 years ago

test_transducer unit test results:

----------------------------------------------------------------------
Ran 17 tests in 5.940s

OK (skipped=6)

ROCm / apex