search

facebookresearch / FAMBench

Benchmarks to capture important workloads.
Apache License 2.0
28 stars 23 forks source link

[ROCm] ROCm Implementation for RNN-T Train #91

Closed jpvillam-amd closed 1 year ago

jpvillam-amd commented 2 years ago

NOTE: To run benchmarks/setup_rnnt_amd.sh use a docker container with rocm/pytorch:latest

amathews-amd commented 2 years ago

@erichan1, @nrsatish , although it seems that we have ability to "Merge pull request" now, could you handle it ?