Commit fa9872f

sarunya

authored and

facebook-github-bot

committed

Enable pipeline prefetching

Differential Revision: D60727327

1 parent 3070f88 commit fa9872fCopy full SHA for fa9872f

5 files changed

+803

-318

lines changed

fbgemm_gpu
- fbgemm_gpu/tbe/ssd
  - training.py
- src/ssd_split_embeddings_cache
  - ssd_scratch_pad_indices_queue.cpp
  - ssd_table_batched_embeddings.h
- test/tbe/ssd
  - ssd_split_tbe_training_test.py
  - ssd_utils_test.py

5 files changed

+803

-318

lines changed

Comments

(0)