pytorch/xla

Pull Requests Commits

Update cuda_driver.diff

ManfeiBai committed 2 years ago

Verified 4a162394

Update WORKSPACE

ManfeiBai committed 2 years ago

Verified 27e4509e

Create cuda_driver.diff

ManfeiBai committed 2 years ago

Verified c9d8a721

Update setup.py

ManfeiBai committed 2 years ago

Verified cc32005b

Update WORKSPACE

ManfeiBai committed 2 years ago

Verified 20733866

[dynamo] Move CPU tensor constructor nodes to XLA. (#5922)

ysiraichi committed 2 years ago

Verified 9cccbe10

Add documentation and python API for persistent cache (#6046)

jonb377 committed 2 years ago

Verified a01de392

benchmarks/aggregate: use typing.Dict[] instead of builtins.dict[] (#6047)

cota committed 2 years ago

Verified 39fcf8bd

benchmarks/aggregate: use typing.List[] instead of builtins.list[] (#6045)

cota committed 2 years ago

Verified e1fe4836

benchmarks/aggregate: import matplotlib only if --output=png (#6044)

cota committed 2 years ago

Verified 7b19f25a

benchmarks/tests: move contents to tests/benchmarks (#6042)

cota committed 2 years ago

Verified d78986d8

Add fallback ops tracking capability. (#6028)

golechwierowicz committed 2 years ago

Verified 53148502

Fix CUDA profiling dump. (#6030)

golechwierowicz committed 2 years ago

Verified 65176ff2

Add a patch for mismatched batch size. (#6029)

golechwierowicz committed 2 years ago

Verified 3dfd0f9b

Add reduce-scatter coalescing for FSDP/ZeRO1 (#5956)

jeffhataws committed 2 years ago

Verified ac94781b

Enable persistent compilation caching (#5804)

jonb377 committed 2 years ago

Verified fae01668

update pytorch version for build (#6038)

zpcore committed 2 years ago

Verified c8fdd61c

benchmarks/aggregate: add tests + use oldest benchmarked Inductor as baseline (#6034)

cota committed 2 years ago

Verified fe4af008

handle dim_size=0 in slice (#6027)

lsy323 committed 2 years ago

Verified beba2248

Make PT_XLA_DEBUG analysis's number of frame printed configurable (#5958)

JackCaoG committed 2 years ago

Verified a4ce808e

benchmarks/result_analyzer: set default output format to csv (#6033)

cota committed 2 years ago

Verified fb690300

Incorporate compilation environment in hash (#5803)

jonb377 committed 2 years ago

Verified 1ea23741

Add a spmd for application to pass in topology (#5931)

amithrm committed 2 years ago

Verified 10618240

Update artifacts.auto.tfvars for 2.2 release (#5908)

zpcore committed 2 years ago

Verified 51df39f3

Add a persistent cache (#5800)

jonb377 committed 2 years ago

Verified f06f6c00

Fix bug with IsTpuDevice for PJRT_DEVICE=TPU_C_API (#6031)

yeounoh committed 2 years ago

Verified 216e07dc

Enable Cumsum with long input (#5974)

JackCaoG committed 2 years ago

Verified 0287a3b5

Enable reduce scatter test on gpu (#6019)

JackCaoG committed 2 years ago

Verified b7cf37af

Disable flaky test (#6021)

JackCaoG committed 2 years ago

Verified 062aa91a

Update libtpu version to Nov24(#6026)

ManfeiBai committed 2 years ago

Verified e2229fd5

Older