Skip to content

Add op statistics dump for woq #1876

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

Merged
merged 6 commits into from
Jun 18, 2024
Merged

Add op statistics dump for woq #1876

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
3c27b95
dump op statistics
Kaihui-intel Jun 18, 2024
8e9e5a7
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 18, 2024
ad8073b
add prettytable into requirements_pt
Kaihui-intel Jun 18, 2024
3541111
clean code
Kaihui-intel Jun 18, 2024
f099c92
remove alias
Kaihui-intel Jun 18, 2024
56ae088
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 18, 2024
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
remove alias 
Signed-off-by: Kaihui-intel <[email protected]>
  • Loading branch information
@Kaihui-intel
Kaihui-intel committed Jun 18, 2024
commit f099c927296ce705202bad3b3a37bff5893a169f
4 changes: 2 additions & 2 deletions neural_compressor/torch/utils/utility.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,9 @@

from typing import Callable, Dict, List, Tuple, Union

import prettytable as pt
import torch
from typing_extensions import TypeAlias
from prettytable import PrettyTable

from neural_compressor.common.utils import LazyImport, Mode, logger

Expand Down Expand Up @@ -182,7 +182,7 @@ def __init__(self, data, header, field_names, output_handle=logger.info):
self.header = header
self.data = data
self.output_handle = output_handle
self.tb = pt.PrettyTable(min_table_width=40)
self.tb = PrettyTable(min_table_width=40)

def print_stat(self):
"""Print the statistics."""
Expand Down