[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:
Hongxin Liu
2023-09-19 14:20:26 +08:00
committed by GitHub
parent 3c6b831c26
commit 079bf3cb26
1268 changed files with 50037 additions and 38444 deletions

View File

@@ -1,12 +1,11 @@
import operator
from typing import Any, List, Union
from typing import Any
import torch
from torch.fx.proxy import Attribute, Proxy
from torch.fx.proxy import Proxy
from colossalai.fx.tracer.meta_patch import meta_patched_function
__all__ = ['ColoProxy']
__all__ = ["ColoProxy"]
class ColoProxy(Proxy):
@@ -39,11 +38,12 @@ class ColoProxy(Proxy):
return self._meta_data is not None
def _assert_meta_data_is_tensor(self):
assert torch.is_tensor(
self._meta_data) and self._meta_data.is_meta, f'Meta data is not a meta tensor for {self.node.name}'
assert (
torch.is_tensor(self._meta_data) and self._meta_data.is_meta
), f"Meta data is not a meta tensor for {self.node.name}"
def _assert_has_meta_data(self):
assert self._meta_data is not None, f'Meta data is not set for {self.node.name}'
assert self._meta_data is not None, f"Meta data is not set for {self.node.name}"
def __len__(self):
self._assert_has_meta_data()
@@ -62,7 +62,6 @@ class ColoProxy(Proxy):
return self.meta_data
def __getattr__(self, k):
return ColoAttribute(self, k)
def __contains__(self, key):
@@ -92,7 +91,6 @@ def extract_meta(*args, **kwargs):
class ColoAttribute(ColoProxy):
def __init__(self, root, attr: str):
self.root = root
self.attr = attr