pytorch/xla

Pull Requests Commits

Update README.md (#9397)

shauheen committed 135 days ago

Verified 08187535

Add jax_device context manager to control the device target (#9382)

zzzwen committed 135 days ago

Verified d780e55c

Refactor call_jax to allow implementing ops in python (#9354)

qihqi committed 135 days ago

Verified 1784a081

[build_developer] Fix vision installation command (#9395)

tengyifei committed 136 days ago

Verified dabcb2dc

Update gru.py to use is_fn_pure (#9393)

tengyifei committed 136 days ago

Verified 86471c07

Update documentations for scan cache (#9394)

iwknow committed 137 days ago

Verified f21f9c0f

Make the shape property adhere to torch.Tensor interface. (#9342)

hfan committed 137 days ago

Verified b4108cb5

Torchax: JittableModule isinstance to work with encapsulated model (#9375)

zmelumian972 committed 137 days ago

Verified 2533c429

Add cache support for scan_layers (#9297)

iwknow committed 137 days ago

Verified 52994af5

Wrap def_static to enable warning reporting. (#9380)

zhanyong-wan committed 137 days ago

Verified 39251247

Revert "Avoid unnecessary copy in TensorSource (#8849)" (#9379)

jeffhataws committed 137 days ago

Verified efdf117b

Implement prng_key as a mutable array (#9305)

tengyifei committed 138 days ago

Verified 0ee876f3

Include both pytorch and torch_xla revisions in the compilation cache key. (#9383)

zhanyong-wan committed 138 days ago

Verified aaff9591

Add autocast feature as `torchax.amp.autocast`. (#9364)

qihqi committed 138 days ago

Verified 89cc066a

Update the tuned block size. (#9376)

QiliangCui committed 138 days ago

Verified f20b89f9

Deprecate ShapeOfXlaOp in favor of GetShape (#9381)

zhanyong-wan committed 138 days ago

Verified b737b4a1

Fix race condition in runAtenTest (#9306)

benawilson committed 138 days ago

Verified f8bd6c48

Improve the API for binding python functions in C++ (#9351)

zhanyong-wan committed 138 days ago

Verified 86c59725

update l_ref in kernel matrix calculation (#9372)

Chenyaaang committed 139 days ago

Verified 77c40f3f

Avoid recompilation caused by `scan_layers` (#9367)

tengyifei committed 139 days ago

Verified 41b9e452

fix all_gather_into_tensor test and logic (#9332)

bfolie committed 139 days ago

Verified 9b0b02f0

Improve `assume_pure` SPMD functionality (#9360)

tengyifei committed 139 days ago

Verified 0be0fbdf

Add an option to not use dlpack. (#9304)

qihqi committed 140 days ago

Verified 55a75404

docs: fix contribute guide command (#9355)

giacomoni committed 141 days ago

Verified 20899c72

Update ragged paged attention kernel to prevent vmem oom (#9346)

Chenyaaang committed 142 days ago

Verified bea86eec

Split _ops and _decomps. (#9323)

qihqi committed 143 days ago

Verified 3a1ed628

Ignore the MODULE.bazel* files generated by bazel. (#9352)

zhanyong-wan committed 143 days ago

Verified 15366e94

Switch XLA_CHECKs to ABSL_CHECKs in lowering_context. (#9338)

zhanyong-wan committed 143 days ago

Verified c2aa21ee

cleanup: remove defunct --distinct_host_configuration flag (#9300)

rickeylev committed 143 days ago

Verified 1d409399

Fix usage of dimensions_size() to check for tuple (#9347)

bfolie committed 143 days ago

Verified 49b0b3b3

Newer Older