strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] some minor fixes

Chillee committed 2 years ago

afba1985

[functorch] [pytree] register return_types (pytorch/functorch#350)

kshitij12345 committed 2 years ago

a8d54b40

[functorch] xfail failing jvp tests

Sam Andow committed 2 years ago

524002be

[functorch] skip to solve nasty seg fault

Sam Andow committed 2 years ago

21bd4ba5

[functorch] Use num_bytes instead of numel in min-cut partitioning (pytorch/functorch#398)

anijain2305 committed 2 years ago

633d1f02

[functorch] group norm + backwards running (pytorch/functorch#387)

Samantha Andow committed 2 years ago

f0f65be5

[functorch] fix some partitioning issues

Chillee committed 2 years ago

e5980eab

[functorch] Cleanup for memory efficient fusion (pytorch/functorch#388)

anijain2305 committed 2 years ago

183e88cd

[functorch] fix remaining failing cuda test

Sam Andow committed 2 years ago

18a374c2

[functorch] cleaned up code and fixed failures

Chillee committed 2 years ago

6e5c2680

[functorch] made some updates to min-cut solver

Chillee committed 2 years ago

a5c7dade

[functorch] fix unexpected success

Sam Andow committed 2 years ago

feb2073a

[functorch] [Compile Cache] Handle non tensor args (pytorch/functorch#383)

anijain2305 committed 2 years ago

83bb9033

[functorch] fix flake8 issues?

Chillee committed 2 years ago

1c666edd

[functorch] binary_cross_entropy batch rule (pytorch/functorch#380)

zou3519 committed 2 years ago

c6a7b94a

[functorch] fix test failures

Chillee committed 2 years ago

bdcc5286

[functorch] fix flake8 issue

Chillee committed 2 years ago

bbf2e80e

[functorch] fix CI test for an op I don't have locally yet :)

Chillee committed 2 years ago

de4b1e2f

[functorch] fix some random minifier and add some random decompositions

Chillee committed 2 years ago

e47b8f33

[functorch] switch per sample grad test to use group norm

Sam Andow committed 2 years ago

e735de57

[functorch] Add Batch Norm (+ backward) batching rules (pytorch/functorch#371)

Samantha Andow committed 2 years ago

ace198ee

[functorch] Fixed errors

Chillee committed 2 years ago

33110e8a

[functorch] fix flake issues

Chillee committed 2 years ago

747c2fa2

[functorch] refactored decompositions a bit

Chillee committed 2 years ago

02ae449c

[functorch] max-flow based algorithm for activation checkpointing (pytorch/functorch#344)

Chillee committed 2 years ago

3b8b72a3

[functorch] made tests that work on either cpu or cuda but not both skips

Sam Andow committed 2 years ago

a8e01d4b

[functorch] fix ci, forgotten commas and use xfail

Sam Andow committed 2 years ago

dd0fe720

[functorch] fix ci by disabling certain tests

Sam Andow committed 2 years ago

761c3825

[functorch] Dropped 3.6 and added 3.9

vfdev-5 committed 2 years ago

0d37b12f

[functorch] Remove make_nnc from test_pythonkey import

zou3519 committed 2 years ago

47f69917

Newer Older