strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] moved some decompositions from batchingregistrations.cpp out

Chillee committed 2 years ago

23fc2e0f

[functorch] added remove-inplace pass to nnc_jit (kinda works, but won't work with aliasing)

Chillee committed 2 years ago

71042b1b

[functorch] Update dependency (pytorch/functorch#58)

zou3519 committed 2 years ago

abc520f8

[functorch] (Grad)TensorWrapper sometimes has storage (and data_ptr) (pytorch/functorch#65)

zou3519 committed 2 years ago

a9ac8e81

[functorch] Install expecttest as CI step (pytorch/functorch#66)

zou3519 committed 2 years ago

26503f21

[functorch] added ability for module attrs to be directly referenced in output

Chillee committed 2 years ago

74f77319

[functorch] Added binary_cross_entropy lowerings

Chillee committed 2 years ago

f10845ee

[functorch] fixed device issues in tests

Chillee committed 2 years ago

379ae35e

[functorch] add no-dim for batchrulesreduce and move trace from old batching rule to new decomposition

Chillee committed 2 years ago

f561f0f6

[functorch] Removed no-dim rules for batchrulespooling and batchrulesviews

Chillee committed 2 years ago

0a8ef33c

[functorch] removed no bdim batching rules from batchrulesbinaryops

Chillee committed 2 years ago

8f06e915

[functorch] removed no bdim cases from batchruleslinearalgebra

Chillee committed 2 years ago

cb286b9b

[functorch] Added a test for case where we would previously have dispatched to a batching rule despite having no bdims

Chillee committed 2 years ago

8cd80e0b

[functorch] Selectively enable dispatch on kBatchedKey (pytorch/functorch#63)

zou3519 committed 2 years ago

446d0b4e

[functorch] Added einsum batching rule

Chillee committed 2 years ago

bbfdcdbd

[functorch] updated opinfo DB and added meshgrid batching rule

Chillee committed 2 years ago

495112f5

[functorch] Exclude failing tests, add xfail test for torch.tensor

zou3519 committed 2 years ago

f6667347

[functorch] Add flip batching rule

Chillee committed 2 years ago

c02cc07c

[functorch] Added batching rule for logsumexp

Chillee committed 2 years ago

b20e4dec

[functorch] Added norm.Scalar and norm.ScalarOpt_dim

Chillee committed 2 years ago

9c138786

[functorch] Revert "Exclude List[Optional[Tensor]] from the batched fallback"

Chillee committed 2 years ago

65c54e28

[functorch] Added full_like and refactored factory macros a bit

Chillee committed 2 years ago

6916cc5d

[functorch] Added triu/tril

Chillee committed 2 years ago

b815b5bc

[functorch] finished batching rules for div

Chillee committed 2 years ago

0aedd9e8

[functorch] Added sort batching rules

Chillee committed 2 years ago

345cf3eb

[functorch] finished off where batching rules + added OP_DECOMPOSE macro

Chillee committed 2 years ago

94a55b7e

[functorch] Added where/_s_where, changed flatten (a composite op) to decomposition, and added (var/std).correction

Chillee committed 2 years ago

7b847833

[functorch] Added erf/inverse/isinf/isnan batching rules

Chillee committed 2 years ago

60fbba9d

[functorch] Added isnan + topk batching rules

Chillee committed 2 years ago

58319030

[functorch] Added clamp.tensor batching rule

Chillee committed 2 years ago

7bb63ff2

Newer Older