mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-09 21:09:18 +00:00
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
This commit is contained in:
@@ -36,6 +36,7 @@ class ColoParamOpHookManager:
|
||||
Manage your param op hooks. It only has static methods.
|
||||
The only static method you should call is ``use_hooks(*hooks)``.
|
||||
"""
|
||||
|
||||
hooks: Tuple[ColoParamOpHook, ...] = tuple()
|
||||
|
||||
@staticmethod
|
||||
@@ -99,7 +100,6 @@ class ColoParamOpHookManager:
|
||||
|
||||
|
||||
class PreFwdPostBwd(torch.autograd.Function):
|
||||
|
||||
@staticmethod
|
||||
def forward(ctx, params, *args):
|
||||
ctx.params = params
|
||||
@@ -112,7 +112,6 @@ class PreFwdPostBwd(torch.autograd.Function):
|
||||
|
||||
|
||||
class PostFwdPreBwd(torch.autograd.Function):
|
||||
|
||||
@staticmethod
|
||||
def forward(ctx, params, args):
|
||||
ctx.params = params
|
||||
|
Reference in New Issue
Block a user