QErrorComputer.TAEComputer Class Reference

Collaboration diagram for QErrorComputer.TAEComputer:

Public Member Functions
	__init__ (self, fp32_dir, fq_dir)
	advance_on (self, fp32_dir, fq_dir)
	get_final_result (self)
	run (self)
Public Member Functions inherited from QErrorComputer.QErrorComputer
	collect_data_path (self, fp32_dir, fq_dir)

Data Fields
	total_error
	qerror_min
	qerror_max
Data Fields inherited from QErrorComputer.QErrorComputer
	qerror_map

Protected Attributes
	_fp32_dir
	_fq_dir
Protected Attributes inherited from QErrorComputer.QErrorComputer
	_fp32_dir
	_fq_dir
	_num_processed_data
	_num_data
	_tid_to_tname

Detailed Description

Definition at line 172 of file QErrorComputer.py.

Constructor & Destructor Documentation

__init__()

QErrorComputer.TAEComputer.__init__	(		self,
			fp32_dir,
			fq_dir
	)

Reimplemented from QErrorComputer.QErrorComputer.

Definition at line 173 of file QErrorComputer.py.

    def __init__(self, fp32_dir, fq_dir):
        super().__init__(fp32_dir, fq_dir)
        self.total_error = 0
        self.qerror_min = float('inf')
        self.qerror_max = -self.qerror_min
 

References QErrorComputer.TAEComputer.__init__().

Referenced by QErrorComputer.TAEComputer.__init__().

Member Function Documentation

advance_on()

QErrorComputer.TAEComputer.advance_on	(		self,
			fp32_dir,
			fq_dir
	)

Definition at line 179 of file QErrorComputer.py.

    def advance_on(self, fp32_dir, fq_dir):
        data_paths = self.collect_data_path(fp32_dir, fq_dir)
        for tensor_name, data_path in data_paths.items():
            for (fp32_data_path, fq_data_path) in data_path:
                fp32_data = np.load(fp32_data_path)
                fq_data = np.load(fq_data_path)
 
                total_error = np.sum(np.abs(fp32_data - fq_data))
 
                self.qerror_map[tensor_name] += total_error
 
                self.qerror_min = min(total_error, self.qerror_min)
                self.qerror_max = max(total_error, self.qerror_max)
 

References QErrorComputer.QErrorComputer.collect_data_path(), QErrorComputer.QErrorComputer.qerror_map, QErrorComputer.MSEComputer.qerror_max, QErrorComputer.TAEComputer.qerror_max, QErrorComputer.MSEComputer.qerror_min, and QErrorComputer.TAEComputer.qerror_min.

get_final_result()

QErrorComputer.TAEComputer.get_final_result

(

self

)

Definition at line 197 of file QErrorComputer.py.

    def get_final_result(self):
        qerror_map = dict()
        for tensor_name, acc in self.qerror_map.items():
            qerror_map[tensor_name] = acc / self._num_processed_data
        return qerror_map, self.qerror_min, self.qerror_max
 

References QErrorComputer.QErrorComputer._num_processed_data, QErrorComputer.QErrorComputer.qerror_map, QErrorComputer.MSEComputer.qerror_max, QErrorComputer.TAEComputer.qerror_max, QErrorComputer.MSEComputer.qerror_min, and QErrorComputer.TAEComputer.qerror_min.

run()

QErrorComputer.TAEComputer.run

(

self

)

Return qerror map (dict: tensor_name(string) -> qerror(float)).

Reimplemented from QErrorComputer.QErrorComputer.

Definition at line 203 of file QErrorComputer.py.

    def run(self):
        self.advance_on(self._fp32_dir, self._fq_dir)
        return self.get_final_result()
 
 
# Scaled Root Mean Square Error (SRMSE)
# SRMSE = sqrt(MSE) / scale

Field Documentation

_fp32_dir

QErrorComputer.TAEComputer._fp32_dir

protected

Definition at line 204 of file QErrorComputer.py.

_fq_dir

QErrorComputer.TAEComputer._fq_dir

protected

Definition at line 204 of file QErrorComputer.py.

qerror_max

QErrorComputer.TAEComputer.qerror_max

Definition at line 177 of file QErrorComputer.py.

Referenced by QErrorComputer.MSEComputer.advance_on(), QErrorComputer.TAEComputer.advance_on(), QErrorComputer.MSEComputer.get_final_result(), and QErrorComputer.TAEComputer.get_final_result().

qerror_min

QErrorComputer.TAEComputer.qerror_min

Definition at line 176 of file QErrorComputer.py.

Referenced by QErrorComputer.MSEComputer.advance_on(), QErrorComputer.TAEComputer.advance_on(), QErrorComputer.MSEComputer.get_final_result(), and QErrorComputer.TAEComputer.get_final_result().

total_error

QErrorComputer.TAEComputer.total_error

Definition at line 175 of file QErrorComputer.py.

---

The documentation for this class was generated from the following file:

• compiler/visq/visqlib/QErrorComputer.py