├── .circleci └── config.yml ├── .flake8 ├── .github └── workflows │ └── pythonpublish.yml ├── .gitignore ├── CONTRIBUTING.md ├── DrQA ├── README.md ├── download.sh ├── drqa │ ├── layers.py │ ├── model.py │ ├── rnn_reader.py │ └── utils.py ├── prepro.py ├── requirements.txt └── train.py ├── LICENSE ├── README.md ├── classification ├── README.md ├── dataloader.py ├── modules.py └── train_classifier.py ├── docs └── pytorch_installation.md ├── imgs ├── classification.png └── speed.png ├── language_model ├── README.md ├── train_enwik8.py └── train_lm.py ├── misc ├── compare_cpu_speed_sru_gru.py ├── compare_gpu_speed_sru_gru.py ├── test_backward_with_transpose.py ├── test_impl.py ├── test_mm.py ├── test_multigpu.py └── test_sru.py ├── requirements-test.txt ├── requirements.txt ├── setup.py ├── speech ├── BLSTM_5L.ndl ├── Bi-SRU_12L.ndl ├── CNTK_Bi-SRU.config ├── CNTK_Bi-SRU_smbr.config ├── CNTK_SRU.config ├── CNTK_SRU_smbr.config ├── LSTM_5L.ndl ├── README.md ├── SRU_12L.ndl ├── make_fbank_pitch.sh ├── run_bi-sru.sh ├── run_sru.sh └── run_swbd.sh ├── sru ├── __init__.py ├── csrc │ ├── CMakeLists.txt │ ├── README.md │ ├── main_test_cpp.cpp │ ├── sru_cpu_impl.cpp │ ├── sru_cuda_impl.cpp │ ├── sru_cuda_impl_dummy.cpp │ └── sru_cuda_kernel.cu ├── cuda_functional.py ├── modules.py ├── ops.py └── version.py └── test ├── regression ├── artifacts │ └── 2.3.5.pt ├── build_artifact.py ├── build_artifact.sh └── test_regression.py ├── sru ├── test_sru.py └── test_torchscript.py ├── test.sh ├── test_amp.py └── test_ts_cpp.py /.circleci/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/.circleci/config.yml -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/pythonpublish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/.github/workflows/pythonpublish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /DrQA/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/README.md -------------------------------------------------------------------------------- /DrQA/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/download.sh -------------------------------------------------------------------------------- /DrQA/drqa/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/drqa/layers.py -------------------------------------------------------------------------------- /DrQA/drqa/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/drqa/model.py -------------------------------------------------------------------------------- /DrQA/drqa/rnn_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/drqa/rnn_reader.py -------------------------------------------------------------------------------- /DrQA/drqa/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/drqa/utils.py -------------------------------------------------------------------------------- /DrQA/prepro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/prepro.py -------------------------------------------------------------------------------- /DrQA/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/requirements.txt -------------------------------------------------------------------------------- /DrQA/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/DrQA/train.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/README.md -------------------------------------------------------------------------------- /classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/classification/README.md -------------------------------------------------------------------------------- /classification/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/classification/dataloader.py -------------------------------------------------------------------------------- /classification/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/classification/modules.py -------------------------------------------------------------------------------- /classification/train_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/classification/train_classifier.py -------------------------------------------------------------------------------- /docs/pytorch_installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/docs/pytorch_installation.md -------------------------------------------------------------------------------- /imgs/classification.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/imgs/classification.png -------------------------------------------------------------------------------- /imgs/speed.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/imgs/speed.png -------------------------------------------------------------------------------- /language_model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/language_model/README.md -------------------------------------------------------------------------------- /language_model/train_enwik8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/language_model/train_enwik8.py -------------------------------------------------------------------------------- /language_model/train_lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/language_model/train_lm.py -------------------------------------------------------------------------------- /misc/compare_cpu_speed_sru_gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/misc/compare_cpu_speed_sru_gru.py -------------------------------------------------------------------------------- /misc/compare_gpu_speed_sru_gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/misc/compare_gpu_speed_sru_gru.py -------------------------------------------------------------------------------- /misc/test_backward_with_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/misc/test_backward_with_transpose.py -------------------------------------------------------------------------------- /misc/test_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/misc/test_impl.py -------------------------------------------------------------------------------- /misc/test_mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/misc/test_mm.py -------------------------------------------------------------------------------- /misc/test_multigpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/misc/test_multigpu.py -------------------------------------------------------------------------------- /misc/test_sru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/misc/test_sru.py -------------------------------------------------------------------------------- /requirements-test.txt: -------------------------------------------------------------------------------- 1 | pytest 2 | flake8 3 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.5.1 2 | ninja 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/setup.py -------------------------------------------------------------------------------- /speech/BLSTM_5L.ndl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/BLSTM_5L.ndl -------------------------------------------------------------------------------- /speech/Bi-SRU_12L.ndl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/Bi-SRU_12L.ndl -------------------------------------------------------------------------------- /speech/CNTK_Bi-SRU.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/CNTK_Bi-SRU.config -------------------------------------------------------------------------------- /speech/CNTK_Bi-SRU_smbr.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/CNTK_Bi-SRU_smbr.config -------------------------------------------------------------------------------- /speech/CNTK_SRU.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/CNTK_SRU.config -------------------------------------------------------------------------------- /speech/CNTK_SRU_smbr.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/CNTK_SRU_smbr.config -------------------------------------------------------------------------------- /speech/LSTM_5L.ndl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/LSTM_5L.ndl -------------------------------------------------------------------------------- /speech/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/README.md -------------------------------------------------------------------------------- /speech/SRU_12L.ndl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/SRU_12L.ndl -------------------------------------------------------------------------------- /speech/make_fbank_pitch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/make_fbank_pitch.sh -------------------------------------------------------------------------------- /speech/run_bi-sru.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/run_bi-sru.sh -------------------------------------------------------------------------------- /speech/run_sru.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/run_sru.sh -------------------------------------------------------------------------------- /speech/run_swbd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/speech/run_swbd.sh -------------------------------------------------------------------------------- /sru/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/__init__.py -------------------------------------------------------------------------------- /sru/csrc/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/csrc/CMakeLists.txt -------------------------------------------------------------------------------- /sru/csrc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/csrc/README.md -------------------------------------------------------------------------------- /sru/csrc/main_test_cpp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/csrc/main_test_cpp.cpp -------------------------------------------------------------------------------- /sru/csrc/sru_cpu_impl.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/csrc/sru_cpu_impl.cpp -------------------------------------------------------------------------------- /sru/csrc/sru_cuda_impl.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/csrc/sru_cuda_impl.cpp -------------------------------------------------------------------------------- /sru/csrc/sru_cuda_impl_dummy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/csrc/sru_cuda_impl_dummy.cpp -------------------------------------------------------------------------------- /sru/csrc/sru_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/csrc/sru_cuda_kernel.cu -------------------------------------------------------------------------------- /sru/cuda_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/cuda_functional.py -------------------------------------------------------------------------------- /sru/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/modules.py -------------------------------------------------------------------------------- /sru/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/sru/ops.py -------------------------------------------------------------------------------- /sru/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '2.6.0' 2 | -------------------------------------------------------------------------------- /test/regression/artifacts/2.3.5.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/regression/artifacts/2.3.5.pt -------------------------------------------------------------------------------- /test/regression/build_artifact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/regression/build_artifact.py -------------------------------------------------------------------------------- /test/regression/build_artifact.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/regression/build_artifact.sh -------------------------------------------------------------------------------- /test/regression/test_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/regression/test_regression.py -------------------------------------------------------------------------------- /test/sru/test_sru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/sru/test_sru.py -------------------------------------------------------------------------------- /test/sru/test_torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/sru/test_torchscript.py -------------------------------------------------------------------------------- /test/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/test.sh -------------------------------------------------------------------------------- /test/test_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/test_amp.py -------------------------------------------------------------------------------- /test/test_ts_cpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/asappresearch/sru/HEAD/test/test_ts_cpp.py --------------------------------------------------------------------------------