mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-04 10:34:41 +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:
@@ -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
|
||||
|
Reference in New Issue
Block a user