-
Notifications
You must be signed in to change notification settings - Fork 515
Pull requests: pytorch/FBGEMM
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
custom allgather support multiple dtypes
cla signed
fb-exported
#3498
opened Dec 11, 2024 by
xw285cornell
Loading…
Update triton version to 3.2.0+35c6c7c6284582b3f41c71c150e11b517acf074a
cla signed
#3497
opened Dec 11, 2024 by
brad-mengchi
Loading…
use RocksDB iterator to read key range from ssd embedding
cla signed
fb-exported
#3495
opened Dec 11, 2024 by
jiayulu
Loading…
skip broken inference test that uses ssd TBE
cla signed
fb-exported
#3494
opened Dec 11, 2024 by
jiayulu
Loading…
Fixed FBGEMM fp8 rowwise for irregular shapes
cla signed
fb-exported
#3491
opened Dec 10, 2024 by
zjing14
Loading…
migrate "jagged_flash_attention"
cla signed
fb-exported
#3490
opened Dec 10, 2024 by
brad-mengchi
Loading…
Optimzed backward pass for ROCm devices
ciflow/rocm
cla signed
fb-exported
module: rocm
#3488
opened Dec 9, 2024 by
q10
Loading…
Migrate
multi_head_jagged_flash_attention
SLL ops to OSS
cla signed
fb-exported
#3485
opened Dec 9, 2024 by
q10
Loading…
Fix grid size overflow in generate_vbe_metadata
cla signed
fb-exported
#3484
opened Dec 9, 2024 by
sryap
Loading…
Optimzed backward pass for ROCm devices (#3367)
ciflow/rocm
cla signed
fb-exported
module: rocm
#3468
opened Dec 6, 2024 by
q10
Loading…
Support config based bound check version via extended modes
cla signed
fb-exported
#3454
opened Dec 4, 2024 by
Fiery
Loading…
Remove setting total_cache_hash_size as buffer
cla signed
fb-exported
#3441
opened Dec 3, 2024 by
aporialiao
Loading…
Use GEMM kernel for KleidiAI to accelerate FP16Benchmark
cla signed
#3440
opened Dec 3, 2024 by
milpuz01
Loading…
Create move TBE to right device, and set Cache Load in TBE class
cla signed
fb-exported
#3438
opened Dec 3, 2024 by
aporialiao
Loading…
Unit test for new move tbe from device/cache_load method
cla signed
fb-exported
#3437
opened Dec 3, 2024 by
aporialiao
Loading…
fix to add f-string format to log
cla signed
fb-exported
#3433
opened Dec 2, 2024 by
varun2784
Loading…
Make check_feature_gate_key PT2 compatible
cla signed
fb-exported
#3426
opened Nov 30, 2024 by
sryap
Loading…
Make check_feature_gate_key PT2 compatible
cla signed
fb-exported
#3425
opened Nov 30, 2024 by
sryap
Loading…
Previous Next
ProTip!
Adding no:label will show everything without a label.