pytorch/xla

Pull Requests Commits

wonjoo-wj committed 3 years ago

c4c2e46c

Codegen clamp_min.Tensor

wonjoo-wj committed 3 years ago

a8eab406

Codegen clamp_max.Tensor

wonjoo-wj committed 3 years ago

1ad227ad

Remove Ger lowering as it is not needed (#3855)

JackCaoG committed 3 years ago

Verified 2d880d15

Codegen op and rename existing op to all_dim (#3856)

wonjoo-wj committed 3 years ago

Verified 45689b9f

Update codegen doc with more example (#3847)

JackCaoG committed 3 years ago

Verified 58d2ba02

added TPU_ML_PLATFORM variable (#3846)

miladm committed 3 years ago

Verified 987d5405

Refactor PjRt device transfers to use `BufferFromHostBuffer` (#3833)

will-cromar committed 3 years ago

Verified 5dfd843c

partially codegen adaptive_avgpool3d and backward (#3790)

JackCaoG committed 3 years ago

Verified 1f154ceb

Add parameter broadcasting to PJRT examples. (#3836)

darisoy committed 3 years ago

Verified 477ca244

Respect default dtype (#3798)

peterbell10 committed 3 years ago

Verified 7b6747df

Revert "Revert "Revert of #3822 "Revert "implementing sym_numel_custom"" (#3826)" (#3835)" (#3837)

Krovatkin committed 3 years ago

Verified 774df50c

Revert "Revert of #3822 "Revert "implementing sym_numel_custom"" (#3826)" (#3835)

Krovatkin committed 3 years ago

Verified 3af08737

remove pin (#3834)

Krovatkin committed 3 years ago

Verified 2c446ff1

Revert of #3822 "Revert "implementing sym_numel_custom"" (#3826)

Krovatkin committed 3 years ago

Verified ce6b404c

Support dim=None for std (#3792)

wonjoo-wj committed 3 years ago

Verified 5d294a9e

Revert "implementing sym_numel_custom" (#3822)

Krovatkin committed 3 years ago

Verified ff2e0ea5

implementing sym_numel_custom (#3784)

Krovatkin committed 3 years ago

Verified b358d985

Revert "Bump mistune from 0.8.4 to 2.0.3 in /docs (#3807)" (#3820)

miladm committed 3 years ago

Verified ab039d30

Revert "Update matmul cpp test to check a different counter (#3154)" (#3814)

wonjoo-wj committed 3 years ago

Verified e5a2dd1b

Bump mistune from 0.8.4 to 2.0.3 in /docs (#3807)

dependabot[bot] committed 3 years ago

Verified c3207fa0

Codegen binary_cross_entropy/backward (#3809)

JackCaoG committed 3 years ago

Verified e81b5296

Codegen relu and relu_ (#3796)

wonjoo-wj committed 3 years ago

Verified 11602fc6

Codegen trunc (#3805)

wonjoo-wj committed 3 years ago

Verified 84a77683

Revert "Revert "Repair `SymIntNodeImpl` to unblock `master` HEAD (#3802)" (#3803)" (#3812)

ezyang committed 3 years ago

Verified 6cc7504a

Revert "Repair `SymIntNodeImpl` to unblock `master` HEAD (#3802)" (#3803)

JackCaoG committed 3 years ago

Verified b3342319

Repair `SymIntNodeImpl` to unblock `master` HEAD (#3802)

miladm committed 3 years ago

Verified 1693c1c7

Full codegen for isnan (#3758)

steventk-g committed 3 years ago

Verified 98bbc9cf

Codegen silu selu (#3780)

JackCaoG committed 3 years ago

Verified 63a85d9b

Revert "Revert "Support torch.mean with dim=None (#3752)" (#3757)" (#3782)

wonjoo-wj committed 3 years ago

Verified 15f12393

Older