Commits

Branch selector

 build_in_parallel

User selector

All users

All time

Commit History

Commits on Feb 24, 2025

• Update README.md: Remove an `export` command

  yuantailing

  authored
  fd28caf

  Copy full SHA for fd28caf

Commits on Feb 21, 2025

• update README.md

  yuantailing

  committed
  f1ee5b1

  Copy full SHA for f1ee5b1

• fix: setup.py develop supports --parallel

  yuantailing

  committed
  d9c3507

  Copy full SHA for d9c3507

Commits on Feb 20, 2025

• build extensions in parallel

  yuantailing

  committed
  f0df092

  Copy full SHA for f0df092

Commits on Feb 5, 2025

• Merge pull request #1875 from nWEIdia/remove_deprecated_export_to_pretty_string

  Show description for b216eee
  eqy

  authored
  b216eee

  Copy full SHA for b216eee

Commits on Jan 15, 2025

• Remove deprecated usage of export_to_pretty_string in apex

  nWEIdia

  committed
  1d887b4

  Copy full SHA for 1d887b4

Commits on Dec 14, 2024

• Traceable LayerNorm (#1864)

  yanboliang

  authored
  73375b3

  Copy full SHA for 73375b3

Commits on Dec 9, 2024

• Traceable RMSNorm part 2: non affine fused rms norm (#1862)

  Show description for 2863aa0
  yanboliang

  authored
  2863aa0

  Copy full SHA for 2863aa0

Commits on Dec 5, 2024

• use mem pool API for NCCL zero-copy (#1863)

  Aidyn-A

  authored
  9470501

  Copy full SHA for 9470501

Commits on Nov 28, 2024

• Traceable RMSNorm (#1861)

  yanboliang

  authored
  ac8214e

  Copy full SHA for ac8214e

Commits on Nov 26, 2024

• [test] update fix of #1859 (#1860)

  eqy

  authored
  a1df804

  Copy full SHA for a1df804

Commits on Nov 23, 2024

• Update distributed_test_base.py (#1859)

  eqy

  authored
  683d71b

  Copy full SHA for 683d71b

Commits on Nov 21, 2024

• Update test_group_norm.py (#1858)

  eqy

  authored
  97d8a6b

  Copy full SHA for 97d8a6b

Commits on Nov 1, 2024

• `Tensor.type()` -> `Tensor.scalar_type()` (#1855)

  Show description for e552ad9
  crcrpar

  authored
  e552ad9

  Copy full SHA for e552ad9

• manual impl when compiling (#1854)

  alexdremov

  authored
  e5e8b89

  Copy full SHA for e5e8b89

Commits on Oct 17, 2024

• Avoid NCCL collective coalescing in distopt when not needed (#1847)

  Show description for 6102d2c
  timmoon10

  authored
  6102d2c

  Copy full SHA for 6102d2c

Commits on Sep 28, 2024

• remove `run_transformer` from default lists (#1830)

  crcrpar

  authored
  e13011f

  Copy full SHA for e13011f

Commits on Aug 30, 2024

• Add Unittest For Distributed Adam With CUDA Graph (#1836)

  Show description for b7a4acc
  

  alpha0422

  and

  crcrpar

  authored
  b7a4acc

  Copy full SHA for b7a4acc

• Reformat grad_output if it's not channels last. (#1837)

  alpha0422

  authored
  c3e4adf

  Copy full SHA for c3e4adf

• Enhance Distributed Fused Adam (#1832)

  Show description for 7d5ecf1
  

  alpha0422

  and

  timmoon10

  authored
  7d5ecf1

  Copy full SHA for 7d5ecf1

• Traceable GroupNorm (#1835)

  Show description for 2d2db12
  

  alpha0422

  and

  crcrpar

  authored
  2d2db12

  Copy full SHA for 2d2db12

Commits on Aug 19, 2024

• add **kwargs to DistributedTestBase._run (#1829)

  xwang233

  authored
  7001836

  Copy full SHA for 7001836

Commits on Aug 17, 2024

• Fix illegal memory access with multi_tensor_apply size above INT_MAX (#1825)

  Show description for 79e3dc4
  gdb

  authored
  79e3dc4

  Copy full SHA for 79e3dc4

Commits on Jul 25, 2024

• Allow Configurable Cache Directory (#1821)

  Show description for 59b80ee
  leimao

  authored
  59b80ee

  Copy full SHA for 59b80ee

Commits on Jul 24, 2024

• Do not monkey-patch class methods when registering distopt pre-forward hooks (#1820)

  Show description for f9f19d5
  timmoon10

  authored
  f9f19d5

  Copy full SHA for f9f19d5

Commits on Jul 2, 2024

• release gil (#1816)

  Show description for 23c1f86
  crcrpar

  authored
  23c1f86

  Copy full SHA for 23c1f86

Commits on Jun 29, 2024

• deprecate uses of torch.cuda.amp (#1813)

  Show description for f8e60c4
  Fuzzkatt

  authored
  f8e60c4

  Copy full SHA for f8e60c4

Commits on Jun 8, 2024

• Use torch.testing.all_close instead of get_max_diff in test_lamb.py (#1806)

  Fuzzkatt

  authored
  7b73b12

  Copy full SHA for 7b73b12

Commits on Apr 26, 2024

• Fix reduce_blocks_into_lanes race condition (#1798)

  Show description for a7de60e
  Fuzzkatt

  authored
  a7de60e

  Copy full SHA for a7de60e

Commits on Apr 24, 2024

• NCCL userbuffer for DP RS in DistOpt (#1797)

  Show description for f3f0492
  

  WanZzzzzz

  and

  Qiyu Wan

  authored
  f3f0492

  Copy full SHA for f3f0492

• Add nccl_allocator for zero-copy user buffer (#1796)

  Show description for 6038fc1
  Aidyn-A

  authored
  6038fc1

  Copy full SHA for 6038fc1

Commits on Apr 19, 2024

• Add 2D Fused RoPE (#1784)

  Show description for b5df1cc
  yaox12

  authored
  b5df1cc

  Copy full SHA for b5df1cc

• move to correct device for v1 state (#1783)

  acphile

  authored
  c5f6b79

  Copy full SHA for c5f6b79

Commits on Mar 12, 2024

• Update test_fused_softmax.py (#1782)

  eqy

  authored
  810ffae

  Copy full SHA for 810ffae

Commits on Feb 8, 2024

• Support scaled optimizer state in distributed Adam optimizer (#1771)

  Show description for b496d85
  timmoon10

  authored
  b496d85

  Copy full SHA for b496d85