Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[MetaSchedule] Fix Cyclic Dependency in PyClass Family #10368

Merged
merged 27 commits into from
Mar 4, 2022
Merged

[MetaSchedule] Fix Cyclic Dependency in PyClass Family #10368

Changes from 1 commit
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
e742723
Fix runner & runner future.
zxybazh Feb 23, 2022
54dc391
Fix builder.
zxybazh Feb 23, 2022
f441188
Fix task scheduler, space generator & search strategy.
zxybazh Feb 24, 2022
4834372
Fix linting.
zxybazh Feb 24, 2022
d205a5d
Ignore Mypy.
zxybazh Feb 24, 2022
5bd3b80
Fix database & bring back 2 useful NIE tests.
zxybazh Feb 25, 2022
1825ee6
Fix comments.
zxybazh Feb 25, 2022
33cf92e
Fix cost model.
zxybazh Feb 25, 2022
5225eb2
Clean imports.
zxybazh Feb 25, 2022
f6ed606
Fix task scheduler.
zxybazh Feb 28, 2022
8ad4c32
Staging changes.
zxybazh Mar 1, 2022
2433947
Fix comments.
zxybazh Mar 1, 2022
110400f
Fix mutators.
zxybazh Mar 2, 2022
1f39659
Fix postproc.
zxybazh Mar 2, 2022
4aa6fe7
Fix schedule rules.
zxybazh Mar 2, 2022
e4d918e
Fix feature extractor.
zxybazh Mar 2, 2022
4510b37
Fix measure callbacks.
zxybazh Mar 2, 2022
9795b76
Minor fix.
zxybazh Mar 2, 2022
98f5c54
Fix linting.
zxybazh Mar 2, 2022
7c664b6
Fix linting & rebase.
zxybazh Mar 2, 2022
65aa431
Fix linting.
zxybazh Mar 2, 2022
c829c17
Fix linting for pylint.
zxybazh Mar 2, 2022
298e6d6
Fix mypy.
zxybazh Mar 2, 2022
48a3965
Fix constructor function.
zxybazh Mar 3, 2022
da035ba
Add unit test.
zxybazh Mar 3, 2022
5899ad7
Get rid of self.handle.
zxybazh Mar 3, 2022
9a0bf36
Fix linting.
zxybazh Mar 3, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix mypy.
  • Loading branch information
@zxybazh
zxybazh committed Mar 2, 2022
commit 298e6d60fb571141f5ec6b3352362adea4c58311
10 changes: 5 additions & 5 deletions python/tvm/meta_schedule/cost_model/random_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ class RandomModel(PyCostModel):
https://numpy.org/doc/stable/reference/random/generated/numpy.random.get_state.html
"""

import numpy as np # pylint: disable=import-outside-toplevel
import numpy as np # type: ignore # pylint: disable=import-outside-toplevel

random_state: Union[Tuple[str, np.ndarray, int, int, float], dict]
path: Optional[str]
Expand All @@ -57,7 +57,7 @@ def __init__(
path: Optional[str] = None,
max_range: Optional[int] = 100,
):
import numpy as np # pylint: disable=import-outside-toplevel
import numpy as np # type: ignore # pylint: disable=import-outside-toplevel

super().__init__()
if path is not None:
Expand All @@ -75,7 +75,7 @@ def load(self, path: str) -> None:
path : str
The file path.
"""
import numpy as np # pylint: disable=import-outside-toplevel
import numpy as np # type: ignore # pylint: disable=import-outside-toplevel

self.random_state = tuple(np.load(path, allow_pickle=True)) # type: ignore

Expand All @@ -87,7 +87,7 @@ def save(self, path: str) -> None:
path : str
The file path.
"""
import numpy as np # pylint: disable=import-outside-toplevel
import numpy as np # type: ignore # pylint: disable=import-outside-toplevel

np.save(path, np.array(self.random_state, dtype=object), allow_pickle=True)

Expand Down Expand Up @@ -126,7 +126,7 @@ def predict(
result : np.ndarray
The predicted running results.
"""
import numpy as np # pylint: disable=import-outside-toplevel
import numpy as np # type: ignore # pylint: disable=import-outside-toplevel

np.random.set_state(self.random_state)
# TODO(@zxybazh): Use numpy's RandState object:
Expand Down