search

facebookresearch / FAMBench

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

DLRM ubench failing, possibly due to FBGEMM API update #111

Closed amathews-amd closed 1 year ago

amathews-amd commented 1 year ago 
Traceback (most recent call last):
  File "dlrm/ubench/dlrm_ubench_train_embeddingbag_driver.py", line 207, in <module>
    global_elap, global_bytes = run_emb(args, run_dataset)
  File "dlrm/ubench/dlrm_ubench_train_embeddingbag_driver.py", line 104, in run_emb
    requests = bench.split_table_batched_embeddings_benchmark.generate_requests(
TypeError: generate_requests() got an unexpected keyword argument 'weights_precision'

due to https://github.com/pytorch/FBGEMM/commit/4c581375

amathews-amd commented 1 year ago

cc: @jianyuh @nrsatish