-
Notifications
You must be signed in to change notification settings - Fork 537
Pull requests: pytorch/FBGEMM
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Patch D66310520 to make it build in OSS
cla signed
fb-exported
module: rocm
#3409
opened Nov 23, 2024 by
q10
Loading…
Add check to ensure that there is enough room in permuted_indices
cla signed
fb-exported
#3403
opened Nov 22, 2024 by
zimin2000
Loading…
[fbgemm_gpu] Re-enable cache tests for ROCm
ciflow/rocm
cla signed
module: rocm
#3380
opened Nov 15, 2024 by
q10
Loading…
Add support for
int32_t
indices in TBE training (2H/N)
cla signed
fb-exported
#3379
opened Nov 15, 2024 by
q10
Loading…
Add support for
int32_t
indices in TBE training (2/N)
cla signed
fb-exported
#3326
opened Nov 5, 2024 by
q10
Loading…
Add template info into generated files
cla signed
fb-exported
#3325
opened Nov 5, 2024 by
q10
Loading…
Update benchmark test for
Int32_t
Indicies
cla signed
fb-exported
#3317
opened Nov 3, 2024 by
q10
Loading…
Unitifed Prefetching API for CPU TBE
cla signed
fb-exported
#3314
opened Nov 2, 2024 by
excelle08
Loading…
Allow registering SSD prefetcher after initiliaztion of the TBE module
cla signed
fb-exported
#3313
opened Nov 2, 2024 by
excelle08
Loading…
Store the prefetching results in a queue in SSD prefetcher, retrieve later in .forward() method
cla signed
fb-exported
#3312
opened Nov 2, 2024 by
excelle08
Loading…
Use
c10::irange
in deeplearning/fbgemm/BUCK +10
cla signed
fb-exported
#3288
opened Oct 29, 2024 by
q10
Loading…
Always load custom op registrations
cla signed
fb-exported
#3265
opened Oct 22, 2024 by
jwfromm
Loading…
Back out "Better Shape Function Registration"
cla signed
fb-exported
#3261
opened Oct 21, 2024 by
jwfromm
Loading…
Convert directory fbcode/deeplearning to use the Ruff Formatter
cla signed
fb-exported
#3242
opened Oct 11, 2024 by
tpolasek
Loading…
convert data type of indices and offsets in int_nbit_split_embedding_codegen_lookup_function
cla signed
fb-exported
#3235
opened Oct 9, 2024 by
mrmiywj
Loading…
Checkpoint
iter
for FBGEMM Adam etc
cla signed
fb-exported
#3234
opened Oct 9, 2024 by
csmiler
Loading…
Rowwise F8F8BF16 GEMMs - Auto-generate kernel library, auto-generated heuristics cache, add to FBGEMM quantize_bench
cla signed
fb-exported
#3210
opened Oct 2, 2024 by
manishucsd
Loading…
use iter to compare with gwd_start_iter
cla signed
fb-exported
#3195
opened Sep 30, 2024 by
csmiler
Loading…
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.