pytorch/xla

Pull Requests Commits

Update xla_lower_util.h

ManfeiBai committed 1 year ago

Verified 5282c494

Update xla_lower_util.cpp

ManfeiBai committed 1 year ago

Verified b17a5860

Update tensor_methods.h

ManfeiBai committed 1 year ago

Verified 5e42369e

Update tensor_methods.cpp

ManfeiBai committed 1 year ago

Verified 75e1c703

Update scan.cpp

ManfeiBai committed 1 year ago

Verified 9fa5cb96

ManfeiBai committed 1 year ago

Verified 07b345eb

Create scan.cpp

ManfeiBai committed 1 year ago

Verified bfd7d843

Update aten_xla_type.cpp

ManfeiBai committed 1 year ago

Verified f9618ea8

Update xla_native_functions.yaml

ManfeiBai committed 1 year ago

Verified 66660de4

Handle dynamo counter change after fake tensor cache (#6308)

JackCaoG committed 1 year ago

Verified d6dc1a01

Address GPU CI coverage timeout (#6306)

yeounoh committed 1 year ago

Verified 7fae2ab1

update release build to use rc8 tag (#6310)

zpcore committed 1 year ago

Verified 1cfd0cf5

Support mixed precision for torch.where (#6303)

qihqi committed 1 year ago

Verified d49fc70d

Fix global_runtime_device_count() for spmd case (#6266)

vanbasten23 committed 1 year ago

Verified 27ae6dc9

fix gpu spmd core dump (#6275)

mars1248 committed 1 year ago

Verified de27258c

Fix some more aten core ops (#6301)

wonjoo-wj committed 1 year ago

Verified 5d0de94b

Correct order of constant vs args (#6300)

qihqi committed 1 year ago

Verified 55177798

Enable Tensor index check in both dimension (#6299)

ManfeiBai committed 1 year ago

Verified 28502884

Track mean relative error across models. (#6250)

golechwierowicz committed 1 year ago

Verified 30c6b01c

Support of implicit broadcasting with unbounded dynamism (#6219)

sdasgup3 committed 1 year ago

Verified 896de17e

[Core Aten] Add and enable tests for aten index_select and logical_and (#6293)

chunnienc committed 1 year ago

Verified 04e1238a

Ignore non-XLA nodes and their direct dependents. (#6170)

ysiraichi committed 1 year ago

Verified 81410785

Set XLA_USE_SPMD for spmd cpp tests. (#6273)

vanbasten23 committed 1 year ago

Verified a728afe8

Fix output type of custom calls while lowering quant/dequant torch op to HLO (#6283)

sdasgup3 committed 1 year ago

Verified 68f47508

Add single graph option as a runnable script. (#6277)

golechwierowicz committed 1 year ago

Verified ddf4be26

Fix some more core aten opset tests (#6284)

wonjoo-wj committed 1 year ago

Verified 9f1afbd3

Update configuration.yaml to include XLA_DISABLE_FUNCTIONALIZATION (#6290)

wonjoo-wj committed 1 year ago

Verified 4b1eaeef

Update example Kaggle notebook to PyTorch/XLA 2.1 (#6287)

will-cromar committed 1 year ago

Verified 0e735dec

Add CPU PJRT plugin for testing (#6253)

will-cromar committed 1 year ago

Verified 2f1334d7

Temporarily disable GPU coverage workflow in CI (#6285)

yeounoh committed 1 year ago

Verified 83b7571a

Older