Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 15, 2024

• Fix scalar torchrec metric to be not persistent (pytorch#1704)

  Show description for 7b4b1dd
  

  Andrew Cui

  authored and

  facebook-github-bot

  committed
  7b4b1dd

  Copy full SHA for 7b4b1dd

Commits on Feb 14, 2024

• Optimize _length_per_key_from_stride_per_key with segment sum csr when appropriate (pytorch#1699)

  Show description for 8761b37
  

  joshuadeng

  authored and

  facebook-github-bot

  committed
  8761b37

  Copy full SHA for 8761b37

• Add tests for different sharding types (pytorch#1673)

  Show description for d23b6c7
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  d23b6c7

  Copy full SHA for d23b6c7

• Add option to pass inverse indices to new KJT after permute

  Show description for 4db2a5f
  

  joshuadeng

  authored and

  facebook-github-bot

  committed
  4db2a5f

  Copy full SHA for 4db2a5f

• Add Scalar Metric to TorchRec

  Show description for 21081e4
  

  Andrew Cui

  authored and

  facebook-github-bot

  committed
  21081e4

  Copy full SHA for 21081e4

Commits on Feb 13, 2024

• add performance benchmark for KeyedTensor.regroup (pytorch#1697)

  Show description for 2558cea
  

  efiks

  authored and

  facebook-github-bot

  committed
  2558cea

  Copy full SHA for 2558cea

• Refactor partitioners to make them more modular (pytorch#1696)

  Show description for a860519
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  a860519

  Copy full SHA for a860519

Commits on Feb 9, 2024

• Remove unused import

  Show description for 8e41b29
  

  yanboliang

  authored and

  facebook-github-bot

  committed
  8e41b29

  Copy full SHA for 8e41b29

• Add uvm to allowed compute kernel for zch (pytorch#1695)

  Show description for eaf7f79
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  eaf7f79

  Copy full SHA for eaf7f79

• Move skipfiles to trace_rules.py

  Show description for 1c47e60
  

  yanboliang

  authored and

  facebook-github-bot

  committed
  1c47e60

  Copy full SHA for 1c47e60

• Fix dtype mismatch on meta (pytorch#1689)

  Show description for 47441cf
  

  ge0405

  authored and

  facebook-github-bot

  committed
  47441cf

  Copy full SHA for 47441cf

Commits on Feb 8, 2024

• Adds threading support in torchrec train pipeline. (pytorch#1694)

  Show description for d6b3da6
  

  fhzhang-Meta

  authored and

  facebook-github-bot

  committed
  d6b3da6

  Copy full SHA for d6b3da6

• fix device type from gpu to cuda (pytorch#1688)

  Show description for 68c08c8
  

  gnahzg

  authored and

  facebook-github-bot

  committed
  68c08c8

  Copy full SHA for 68c08c8

• Seed Luus Jaakola with left cost since we already know it. (pytorch#1692)

  Show description for 5bfbeab
  

  Damian Reeves

  authored and

  facebook-github-bot

  committed
  5bfbeab

  Copy full SHA for 5bfbeab

• Switch EmbeddingOffloadScaleupProposer to use Luus Jaakola search (pytorch#1691)

  Show description for 8d93bc2
  

  Damian Reeves

  authored and

  facebook-github-bot

  committed
  8d93bc2

  Copy full SHA for 8d93bc2

• Implement clamped Luus Jaakola search for function minimization. (pytorch#1690)

  Show description for 4696b8d
  

  Damian Reeves

  authored and

  facebook-github-bot

  committed
  4696b8d

  Copy full SHA for 4696b8d

• Improve msg error thrown when no sharding option found (pytorch#1693)

  Show description for 90b72db
  

  sarckk

  authored and

  facebook-github-bot

  committed
  90b72db

  Copy full SHA for 90b72db

• Turn on script and sharded output comparision (pytorch#1687)

  Show description for b617cf8
  

  gnahzg

  authored and

  facebook-github-bot

  committed
  b617cf8

  Copy full SHA for b617cf8

• Add gpu gloo tests (pytorch#1668)

  Show description for 3f1f29f
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  3f1f29f

  Copy full SHA for 3f1f29f

Commits on Feb 6, 2024

• Optimize permuting values for VBE with keyed_jagged_index_select (pytorch#1682)

  Show description for f934ec9
  

  joshuadeng

  authored and

  facebook-github-bot

  committed
  f934ec9

  Copy full SHA for f934ec9

• Allow passing feature names through ParameterConstraints and ShardingOption (pytorch#1686)

  Show description for 7edcb2d
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  7edcb2d

  Copy full SHA for 7edcb2d

• Add fused_params to mch sharders (pytorch#1649)

  Show description for 0c2462f
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  0c2462f

  Copy full SHA for 0c2462f

• Add more comments in sharded_tbes_weights_spec (pytorch#1596)

  Show description for 4f234a2
  

  gnahzg

  authored and

  facebook-github-bot

  committed
  4f234a2

  Copy full SHA for 4f234a2

• fix lints all() usages to short circuit (pytorch#1683)

  Show description for fed5202
  

  joshuadeng

  authored and

  facebook-github-bot

  committed
  fed5202

  Copy full SHA for fed5202

Commits on Feb 5, 2024

• Add percentage based threshold function and unit tests (pytorch#1679)

  Show description for 02771aa
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  02771aa

  Copy full SHA for 02771aa

• conditionally convert model outputs containing session id for NDCG computation (pytorch#1678)

  Show description for f8f6f61
  

  iamzainhuda

  authored and

  facebook-github-bot

  committed
  f8f6f61

  Copy full SHA for f8f6f61

Commits on Feb 2, 2024

• Add flushing and reset_cache_states to pre-hook and hook of state_dict and load_state_dict (pytorch#1674)

  Show description for 73ad8b8
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  73ad8b8

  Copy full SHA for 73ad8b8

Commits on Feb 1, 2024

• add default back for session id in NDCG (pytorch#1677)

  Show description for cda48a6
  

  iamzainhuda

  authored and

  facebook-github-bot

  committed
  cda48a6

  Copy full SHA for cda48a6

• register_buffer for DLRM.triu_indices to make it exportable (pytorch#1670)

  Show description for fc7e83d
  

  wilson100hong

  authored and

  facebook-github-bot

  committed
  fc7e83d

  Copy full SHA for fc7e83d

• Deepcopy FP module even if on meta device (pytorch#1676)

  Show description for 584bb44
  

  s4ayub

  authored and

  facebook-github-bot

  committed
  584bb44

  Copy full SHA for 584bb44

• serialize warmup deply sessions (pytorch#1675)

  Show description for 58cc035
  

  Zheng Yan

  authored and

  facebook-github-bot

  committed
  58cc035

  Copy full SHA for 58cc035

Commits on Jan 31, 2024

• Fix pre-commit (pytorch#1672)

  Show description for f960c8d
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  f960c8d

  Copy full SHA for f960c8d

• Refactor model_parallel tests to allow different (device, backend) combination (pytorch#1667)

  Show description for 8b8eeab
  

  henrylhtsang

  authored and

  facebook-github-bot

  committed
  8b8eeab

  Copy full SHA for 8b8eeab

• Fix forward for D51601041 (pytorch#1671)

  Show description for 5fc2dfa
  

  duduyi2013

  authored and

  facebook-github-bot

  committed
  5fc2dfa

  Copy full SHA for 5fc2dfa

Commits on Jan 30, 2024

• dynamically set session_key for session_ids in NDCG (pytorch#1658)

  Show description for 2005d7f
  

  iamzainhuda

  authored and

  facebook-github-bot

  committed
  2005d7f

  Copy full SHA for 2005d7f

Pagination

Previous

Next