aa2e609136
[LINT] Modernize lint to use pre-commit hooks ( #18807 )
...
This PR migrates existing lint to use pre-commit hooks
2026-02-22 11:03:21 -05:00
d35b858ceb
[CUDNN] Support gradient kernels ( #9986 )
...
* Dgrad nchw, nhwc, fp16 working
commit 426e5dca446a27da49270f45171b58f1bfa21fa9
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 11:48:53 2022 +0900
black
commit 211a58b80f4d0f0b5b0230720e41f35e50cb1eaf
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 11:43:52 2022 +0900
fp16 also works
commit c2a34d473b063873628bff00e51a44cd8e4d0e4f
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 11:36:36 2022 +0900
nhwc test also worked
commit c0609ab147fef30c230a94d16b6c1ba35f7dd9c0
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 11:21:23 2022 +0900
nchw test worked
commit 2bf68c72763708151e9f49f09916a210b2547be8
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 10:41:35 2022 +0900
add test stub
commit c86b1288d5e371f12cba4e1b1866966cb9264401
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 10:32:09 2022 +0900
add python definition stub
commit 3166952f9673376801bf4b5b39eeb6f89452f30a
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 06:57:18 2022 +0900
bwd filter compiled
commit e311ba3d05c5f9424ecb952cb5a520ce81a0828a
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 06:27:55 2022 +0900
dgrad compiled
commit 47f35beb5eeeb7cbf9f6ec7cf8f5c80c65e8da46
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Tue Jan 18 06:16:43 2022 +0900
add dgrad stub
commit ebed032d15b1c3895f541c46ce5d80b6dd769034
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Mon Jan 17 17:01:56 2022 +0900
cpplint
commit 834f54a8c13512130e7d91ca0f54268dc06c5481
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Mon Jan 17 16:55:58 2022 +0900
remove cudnn get output
commit dcbd9c95fdb8ffef9db9c2350430b270461a31c3
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Mon Jan 17 16:28:07 2022 +0900
more refactor
commit 146464e8496fff972bdb1687c4e9d432fe3278d5
Author: Masahiro Masuda <masahi129@gmail.com >
Date: Mon Jan 17 15:57:35 2022 +0900
Introduce SetConvdescriptors to refactor cudnn/conv_forward.cc
* add python function for cudnn wgrad
* adding wgrad test
* black
* wgrad nchw and nhwc worked
* remove bwd algo name stuff
* compute output shape properly
* swap arg order in wgrad
* add kernel size arg in test
* black
* cleanup
* more fix
* fix dgrad test
* support running relay conv2d_backward_weight directly with cudnn
* black
* refactor reference function to support nhwc
* removed unused function
* lint
* enable offloading conv2d_transpose to cudnn dgrad
* relax tol
* name fix, remove print
2022-01-23 06:58:31 +09:00
3ad7c4a769
[Conv2DTransposed] Fix wrong shape check and add new TOPI module to support groups ( #9465 )
...
* f wrong type check in conv2d_transpose
* add test case for conv2d transpose
* add groups support for conv2d_transpose
* add naive implementation and schedule for conv2d with groups
* enable tests for cpu and arm_cpu, raise error for cuda platform
* revert the cuda and generic strategy
* revert back the x86 strategy
* revert back the arm_cpu strategy
* revert back the arm_cpu strategy
* revert back the arm_cpu strategy
* fix EOF of x86
* clang lint updated c++ code
* update topi implementation
* Revert test
* Revert test
* add generic/x86/arm specialization for conv2d_transpose with groups > 1
* remove commentted codes
* fix lint
* fix lint
* fix c++ lint
* fix lint
* fix python lint
* remove comments and reformat
* lint file
* lint code
* fix lint
* update logging information in convolution.h
Co-authored-by: Alicja Kwasniewska <alicja.kwasniewska@sima.ai >
2021-11-12 09:05:39 -08:00
c8064b3ca6
[REFACTOR] Remainings of util => utils ( #6778 )
2020-10-29 13:46:31 -04:00
f13fed55cf
[Format] Convert all Python code w/o CI ( #6448 )
...
* Add black setup
* Tweak pyproject.toml
* Fix syntax issues
* Fix
* Tweak
* Black all Python code
2020-09-11 22:17:24 +09:00
0a4ee30da7
[REFACTOR] topi -> tvm/topi ( #6186 )
...
This PR migrates the topi library as a sub namespace of tvm.
2020-08-02 09:29:29 -07:00
Tianqi Chen