pytorch/xla

Pull Requests Commits

Update test_stablehlo_save_load.py

ManfeiBai committed 2 years ago

Verified 7319d1b4

Update stablehlo.py

ManfeiBai committed 2 years ago

Verified 97c5f718

Update stablehlo.py

ManfeiBai committed 2 years ago

Verified 29ba15bb

Update stablehlo.py

ManfeiBai committed 2 years ago

Verified 4ed77765

Update stablehlo.py

ManfeiBai committed 2 years ago

Verified 4ea3efb0

ManfeiBai committed 2 years ago

Verified f796b322

Update test_stablehlo_save_load.py

ManfeiBai committed 2 years ago

Verified f3f95e8b

Update stablehlo.py

ManfeiBai committed 2 years ago

Verified c5abcc19

Update test_stablehlo_save_load.py

ManfeiBai committed 2 years ago

Verified 43fcd537

Update stablehlo.py

ManfeiBai committed 2 years ago

Verified e5d08878

Update stablehlo.py

ManfeiBai committed 2 years ago

Verified 02b23a22

ManfeiBai committed 2 years ago

Verified ed949d4f

Update stablehlo.py

ManfeiBai committed 2 years ago

Verified eb521245

use a common random seed for test core aten ops (#5949)

qihqi committed 2 years ago

Verified f6a775c4

Distribute Literal->Tensor copies across thread pool (#5825)

jonb377 committed 2 years ago

Verified ec54fd44

Add stack frame id metadata and correct filename and line number for custom op_name (#5838)

mrnikwaws committed 2 years ago

Verified 5fa89223

Update troubleshooting doc with few common env var combination (#5929)

JackCaoG committed 2 years ago

Verified 6c5b7b8e

Add graph hash and num input/output to PT_XLA_DEBUG (#5947)

JackCaoG committed 2 years ago

Verified f3b75ba1

Refactor ExecuteReplicated to operate on sharded data directly (#5737)

will-cromar committed 2 years ago

Verified 7b92a949

Error when changing `PJRT_DEVICE` after runtime initialized (#5948)

will-cromar committed 2 years ago

Verified 346ef7c1

Add benchmark noise reducing info. (#5924)

golechwierowicz committed 2 years ago

Verified c8ce0620

Add link to work log issue in FIX_LOWERING_FOR_CORE_ATEN_OPS.md (#5944)

wonjoo-wj committed 2 years ago

Verified 09f2a14c

Parallelize d2h transfers (#5824)

jonb377 committed 2 years ago

Verified b9475d9e

Update pjrt.md (#5941)

Liyang90 committed 2 years ago

Verified eb728a89

Truncate python stack when outputting frame that cause the graph executation (#5933)

JackCaoG committed 2 years ago

Verified a5e07389

Don't fallback for pow (#5919)

qihqi committed 2 years ago

Verified 402166be

Lower quant/dequant torch op to StableHLO (#5763)

lsy323 committed 2 years ago

Verified a3b0c6ec

Fix `as_strided` for inputs smaller than the arguments specification. (#5914)

ysiraichi committed 2 years ago

Verified 4ac255ae

Collect CUDA/CPU profiling info into result sheets. (#5921)

golechwierowicz committed 2 years ago

Verified c03afb14

Fix get_gpu_name returning empty name. (#5925)

golechwierowicz committed 2 years ago

Verified 16a40cc6

Older