Commits

Branch selector

 main

User selector

sryap

Datepicker

All time

Commit History

Commits on Jun 4, 2024

• Remove D->H sync in SSD-TBE (pytorch#2672)

  Show description for 168ae4c
  

  sryap

  authored and

  facebook-github-bot

  committed
  168ae4c

  Copy full SHA for 168ae4c

• Add --export-trace option in ssd_training (pytorch#2671)

  Show description for 9a8e289
  

  sryap

  authored and

  facebook-github-bot

  committed
  9a8e289

  Copy full SHA for 9a8e289

• Record functions in SSD-TBE training's compute_cache_ptrs (pytorch#2670)

  Show description for 08a0c22
  

  sryap

  authored and

  facebook-github-bot

  committed
  08a0c22

  Copy full SHA for 08a0c22

Commits on May 31, 2024

• Move SSD unit test into test/tbe/ssd (pytorch#2647)

  Show description for 6546246
  

  sryap

  authored and

  facebook-github-bot

  committed
  6546246

  Copy full SHA for 6546246

• Work around masked_select for numel > MAX_INT (pytorch#2648)

  Show description for 0fad083
  

  sryap

  authored and

  facebook-github-bot

  committed
  0fad083

  Copy full SHA for 0fad083

• Create an abstract class for EmbeddingKVDB (pytorch#2642)

  Show description for 5e6c487
  

  sryap

  authored and

  facebook-github-bot

  committed
  5e6c487

  Copy full SHA for 5e6c487

• Add feature table map support (pytorch#2640)

  Show description for cdccc7b
  

  sryap

  authored and

  facebook-github-bot

  committed
  cdccc7b

  Copy full SHA for cdccc7b

• Add FP16 weight and output support (pytorch#2638)

  Show description for 5f9f877
  

  sryap

  authored and

  facebook-github-bot

  committed
  5f9f877

  Copy full SHA for 5f9f877

• Fix SSD TBE backend (pytorch#2645)

  Show description for 8819504
  

  sryap

  authored and

  facebook-github-bot

  committed
  8819504

  Copy full SHA for 8819504

Commits on May 28, 2024

• Add TBE tests to failures_dict_fast.json (pytorch#2632)

  Show description for 9f1d0ef
  

  sryap

  authored and

  facebook-github-bot

  committed
  9f1d0ef

  Copy full SHA for 9f1d0ef

Commits on May 24, 2024

• Add cache conflict miss support (pytorch#2596)

  Show description for db4d379
  

  sryap

  authored and

  facebook-github-bot

  committed
  db4d379

  Copy full SHA for db4d379

Commits on May 18, 2024

• Fix BC issues for get_unique_indices (pytorch#2602)

  Show description for 19a91b9
  

  sryap

  authored and

  facebook-github-bot

  committed
  19a91b9

  Copy full SHA for 19a91b9

Commits on May 15, 2024

• Add helper ops to support cache conflict misses (pytorch#2571)

  Show description for 56d21a0
  

  sryap

  authored and

  facebook-github-bot

  committed
  56d21a0

  Copy full SHA for 56d21a0

Commits on May 6, 2024

• Refactor TBE SSD training tests (pytorch#2485)

  Show description for b83460f
  

  sryap

  authored and

  facebook-github-bot

  committed
  b83460f

  Copy full SHA for b83460f

Commits on Apr 22, 2024

• Fix linter error in torch_type_utils.py (pytorch#2528)

  Show description for 31bdf97
  

  sryap

  authored and

  facebook-github-bot

  committed
  31bdf97

  Copy full SHA for 31bdf97

Commits on Apr 20, 2024

• Add BF16 memoment1/2 in PARTIAL_ROWWISE_ADAM TBE (pytorch#2524)

  Show description for 15b4fd3
  

  sryap

  authored and

  facebook-github-bot

  committed
  15b4fd3

  Copy full SHA for 15b4fd3

• Add BF16 memoment1/2 in PARTIAL_ROWWISE_ADAM TBE (pytorch#2522)

  Show description for 0aa244b
  

  sryap

  authored and

  facebook-github-bot

  committed
  0aa244b

  Copy full SHA for 0aa244b

• Add backend for supporting many optim state types (pytorch#2521)

  Show description for 41aff78
  

  sryap

  authored and

  facebook-github-bot

  committed
  41aff78

  Copy full SHA for 41aff78

Commits on Apr 19, 2024

• Add OptimItem in TBE codegen (pytorch#2518)

  Show description for 9180dd4
  

  sryap

  authored and

  facebook-github-bot

  committed
  9180dd4

  Copy full SHA for 9180dd4

Commits on Apr 18, 2024

• Add GQA attention in experimental/gen_ai (pytorch#2504)

  Show description for e839246
  

  sryap

  authored and

  facebook-github-bot

  committed
  e839246

  Copy full SHA for e839246

Commits on Apr 17, 2024

• Fix the hang issue in some TBE GPU optimizers (pytorch#2509)

  Show description for 3f4f98f
  

  sryap

  authored and

  facebook-github-bot

  committed
  3f4f98f

  Copy full SHA for 3f4f98f

Commits on Apr 3, 2024

• Avoid atomic add in ssd_cache_actions_insert_kernel (pytorch#2471)

  Show description for 712286c
  

  sryap

  authored and

  facebook-github-bot

  committed
  712286c

  Copy full SHA for 712286c

Commits on Apr 2, 2024

• Fix TBE's generate_requests for Zipf when L = 0 (pytorch#2473)

  Show description for 77c4950
  

  sryap

  authored and

  facebook-github-bot

  committed
  77c4950

  Copy full SHA for 77c4950

Commits on Mar 28, 2024

• Enable support for max embedding dim = 2048 in TBE (pytorch#2460)

  Show description for a177d62
  

  sryap

  authored and

  facebook-github-bot

  committed
  a177d62

  Copy full SHA for a177d62

• Add SSD TBE training benchmark (pytorch#2464)

  Show description for 1d3bf45
  

  sryap

  authored and

  facebook-github-bot

  committed
  1d3bf45

  Copy full SHA for 1d3bf45

Commits on Mar 27, 2024

• Limit TBE defused optimizer's max emb dim support to 1024 (pytorch#2458)

  Show description for 30cd70d
  

  sryap

  authored and

  facebook-github-bot

  committed
  30cd70d

  Copy full SHA for 30cd70d

• Use TBE v2 in forward when max_D > 1024 (pytorch#2454)

  Show description for a9ace9c
  

  sryap

  authored and

  facebook-github-bot

  committed
  a9ace9c

  Copy full SHA for a9ace9c

Commits on Mar 26, 2024

• Make number of TBE grad_indices_weights templates constant w.r.t emb dim (pytorch#2452)

  Show description for c67c93b
  

  sryap

  authored and

  facebook-github-bot

  committed
  c67c93b

  Copy full SHA for c67c93b

• Make number of backward templates constant w.r.t emb dim (pytorch#2448)

  Show description for 4665e67
  

  sryap

  authored and

  facebook-github-bot

  committed
  4665e67

  Copy full SHA for 4665e67

Commits on Mar 18, 2024

• Add alias for Vec4T<at::acc_type<T, true>> (pytorch#2434)

  Show description for 9463880
  

  sryap

  authored and

  facebook-github-bot

  committed
  9463880

  Copy full SHA for 9463880

Commits on Mar 15, 2024

• Refactor TBE GPU backward codegen (pytorch#2431)

  Show description for 0049a2c
  

  sryap

  authored and

  facebook-github-bot

  committed
  0049a2c

  Copy full SHA for 0049a2c

• Add VBE testing in cache pipeline prefetching tests (pytorch#2432)

  Show description for d3e1131
  

  sryap

  authored and

  facebook-github-bot

  committed
  d3e1131

  Copy full SHA for d3e1131

• Add VBE testing in linearize_cache_indices test (pytorch#2392)

  Show description for 68adb76
  

  sryap

  authored and

  facebook-github-bot

  committed
  68adb76

  Copy full SHA for 68adb76

Commits on Mar 14, 2024

• Refactor linearize_cache_indices test (pytorch#2389)

  Show description for 234d13d
  

  sryap

  authored and

  facebook-github-bot

  committed
  234d13d

  Copy full SHA for 234d13d

• Add VBE support in UVM caching training (pytorch#2391)

  Show description for 34deb23
  

  sryap

  authored and

  facebook-github-bot

  committed
  34deb23

  Copy full SHA for 34deb23

Pagination

Previous

Next