pytorch/xla

Pull Requests Commits

Update setup.py

ManfeiBai committed 2 years ago

Verified 0be90726

Add option to dump HLOs during benchmark runs (#6200)

frgossen committed 2 years ago

Verified 61d42ebe

experiment_runner: revamp options to dump PyTorch/XLA metrics (#6186)

frgossen committed 2 years ago

Verified ad7c1412

benchmarks/aggregate: support SVG output (#6205)

cota committed 2 years ago

Verified bbeb1b11

benchmarks/aggregate: fix --filter breakage (#6206)

cota committed 2 years ago

Verified e307f1ac

Add default v5 flags (#6168)

jonb377 committed 2 years ago

Verified 2e6e183e

Rename `TransferToServer` to `TransferToDevice` (#6202)

will-cromar committed 2 years ago

Verified 5ac722a1

Dynamic PJRT plugin registration API (#5644)

will-cromar committed 2 years ago

Verified ad14582b

benchmarks/aggregate: report stddev (#6192)

cota committed 2 years ago

Verified 6f3f38f3

Unify the way in which result files are dumped (#6162)

frgossen committed 2 years ago

Verified 63455e0c

Improve precision of rsub; enable test for split copy (#6190)

qihqi committed 2 years ago

Verified 1741e8d5

Parse log level as an enum (#6188)

frgossen committed 2 years ago

Verified d6cabc5a

fix aten.var.correction test case with function wrapper (#6194)

zpcore committed 2 years ago

Verified be37bd27

benchmarks/aggregate: read JSONL files directly (#6191)

cota committed 2 years ago

Verified bd2ec6f8

Fix Inductor runs. (#6195)

golechwierowicz committed 2 years ago

Verified fba48364

Dump pytorch profiles and traces per repeat iteration (#6156)

frgossen committed 2 years ago

Verified 751d6874

Catch exception from AMP mismatch. (#6177)

golechwierowicz committed 2 years ago

Verified 30458aaa

benchmarks: save timestamp in json output (#6163)

cota committed 2 years ago

Verified 3105be6e

benchmarks/aggregate: explicitly pass input files as non-optional args (#6181)

cota committed 2 years ago

Verified c8738110

Fix benchmark test (#6189)

frgossen committed 2 years ago

Verified 9d2429b8

Introduce FSDPv2 (#6122)

alanwaketan committed 2 years ago

Verified 4fe9fe7e

Update artifacts.auto.tfvars (#6176)

ManfeiBai committed 2 years ago

Verified 438c7236

Fix benchmarking for non-dynamo path (#6179)

frgossen committed 2 years ago

Verified 68e48db3

Forward subprocess stdout and stderr in all cases (#6154)

frgossen committed 2 years ago

Verified 5c00aad9

IFRT prototype (#5677)

will-cromar committed 2 years ago

Verified 85b3cdc4

Re-land: Fix model initialization. (#6182)

ysiraichi committed 2 years ago

Verified 38e36443

Promote int to float for tanh operation (consistent with Pytorch) (#6166)

ManfeiBai committed 2 years ago

Verified e123f924

[Core Aten] Enable test_aten_minimum_2 (#6175)

chunnienc committed 2 years ago

Verified 8a0b2eac

benchmarks/aggregate: distinguish between openxla/_eval (#6159)

cota committed 2 years ago

Verified 0c1c68c3

Apply precision config env vars in the root process. (#6152)

golechwierowicz committed 2 years ago

Verified dfcf306e

Older