Collaboration diagram for QErrorComputer.MSEComputer:

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
	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 135 of file QErrorComputer.py.

Constructor & Destructor Documentation

◆ init()

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

Reimplemented from QErrorComputer.QErrorComputer.

Definition at line 136 of file QErrorComputer.py.

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

References QErrorComputer.MSEComputer.__init__().

Referenced by QErrorComputer.MSEComputer.__init__().

Member Function Documentation

◆ advance_on()

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

Definition at line 142 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)
 
                MSE = np.square(fp32_data - fq_data).mean()
 
                self.qerror_map[tensor_name] += MSE
 
                self.qerror_min = min(MSE, self.qerror_min)
                self.qerror_max = max(MSE, 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.MSEComputer.get_final_result ( self )

Definition at line 160 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.MSEComputer.run ( self )

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

Reimplemented from QErrorComputer.QErrorComputer.

Definition at line 167 of file QErrorComputer.py.

    def run(self):
        self.advance_on(self._fp32_dir, self._fq_dir)
        return self.get_final_result()
 
 

Field Documentation

◆ _fp32_dir

QErrorComputer.MSEComputer._fp32_dir

protected

Definition at line 168 of file QErrorComputer.py.

◆ _fq_dir

QErrorComputer.MSEComputer._fq_dir

protected

Definition at line 168 of file QErrorComputer.py.

◆ qerror_max

QErrorComputer.MSEComputer.qerror_max

Definition at line 139 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.MSEComputer.qerror_min

Definition at line 138 of file QErrorComputer.py.