mirror of https://github.com/llvm/torch-mlir
13b9fd62c6
This commit adds a test module specifically for table batch embedding algorithm. This test case is in reference to the FBGEMM table batch embedding: https://github.com/pytorch/FBGEMM/blob/main/fbgemm_gpu/bench/split_table_batched_embeddings_benchmark.py#L270 Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com> |
||
---|---|---|
.. | ||
torchscript |