fix dlrm/ubench/dlrm_ubench_train_embeddingbag_driver.py - Githubissues

facebookresearch / FAMBench

Benchmarks to capture important workloads.

Apache License 2.0

28 stars 23 forks source link

fix dlrm/ubench/dlrm_ubench_train_embeddingbag_driver.py #112

Closed liligwu closed 1 year ago

liligwu commented 1 year ago

fix dlrm/ubench/dlrm_ubench_train_embeddingbag_driver.py due to the API change in FBGEMM. https://github.com/pytorch/FBGEMM/commit/4c581375 Fixes https://github.com/facebookresearch/FAMBench/issues/111