squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add simple hash and eq methods for gemm_operations. (#1053)

Browse Source
This commit is contained in:
Ying Zhang 2023-08-27 17:41:57 -07:00 committed by GitHub
parent 6673df0e48
commit 3a8f57a3c8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/library/scripts/gemm_operation.py
View File

@ -212,6 +212,11 @@ class GemmOperation:
''' The full procedural name indicates architecture, extended name, tile size, and layout. ''' ''' The full procedural name indicates architecture, extended name, tile size, and layout. '''
return self.procedural_name() return self.procedural_name()
def __hash__(self):
return hash(self.configuration_name())
def __eq__(self, other):
return self.configuration_name() == other.configuration_name()
################################################################################################### ###################################################################################################
# #