Huawei_Technology
/
MindSpore-AKG
Not watched
1
0
Fork 0
Code
Releases 1 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: efe42ee8fe
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'efe42ee8fe'
${ noResults }
MindSpore-AKG/tests/ut/python/pass/test_insn_info.py

166 lines
6.6 kB
Raw Blame History 

  1. # Copyright 2019 Huawei Technologies Co., Ltd

  2. #

  3. # Licensed under the Apache License, Version 2.0 (the "License");

  4. # you may not use this file except in compliance with the License.

  5. # You may obtain a copy of the License at

  6. #

  7. # http://www.apache.org/licenses/LICENSE-2.0

  8. #

  9. # Unless required by applicable law or agreed to in writing, software

  10. # distributed under the License is distributed on an "AS IS" BASIS,

  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  12. # See the License for the specific language governing permissions and

  13. # limitations under the License.

  14. 

  15. import akg.tvm

  16. from akg.tvm._ffi.node import NodeBase, register_node

  17. import akg.backend as cce

  18. 

  19. 

  20. @register_node

  21. class StmtStoreInfo(NodeBase):

  22.     def __init__(self, comInfo):

  23. 

  24.         strides = comInfo['strides']

  25.         shape = comInfo['shape']

  26.         var = comInfo['var']

  27.         name = comInfo['name']

  28.         index = comInfo['index']

  29.         dtype = comInfo['dtype']

  30. 

  31.         scope = ""

  32.         if 'scope' in comInfo:

  33.             scope = comInfo['scope']

  34.         elem_offset = akg.tvm.const(0, 'int32')

  35.         if 'elem_offset' in comInfo:

  36.             elem_offset = comInfo['elem_offset']

  37.         insn_offset = akg.tvm.const(0, 'int32')

  38.         if 'insn_offset' in comInfo:

  39.             insn_offset = comInfo['insn_offset']

  40.         data_alignment = -2

  41.         if 'data_alignment' in comInfo:

  42.             data_alignment = comInfo['data_alignment']

  43.         data = None

  44.         if 'data' in comInfo:

  45.             data = comInfo['data']

  46. 

  47.         f = akg.tvm.get_global_func("cce_util.create_storeinfo")

  48.         self.__init_handle_by_constructor__(f, strides, shape, var, scope, name, index, elem_offset, insn_offset, dtype, data_alignment, data)

  49. 

  50. 

  51. def DicttoStmtStoreInfo(comInfo):

  52.     stmtInfo = StmtStoreInfo(comInfo)

  53. 

  54.     return stmtInfo

  55. 

  56. 

  57. def createComInfo(name, strides, shape, var, index, dtype):

  58.     comInfo = {}

  59.     comInfo['name'] = name

  60.     comInfo['strides'] = strides

  61.     comInfo['shape'] = shape

  62.     comInfo['var'] = var

  63.     comInfo['index'] = index

  64.     comInfo['dtype'] = dtype

  65. 

  66.     return DicttoStmtStoreInfo(comInfo)

  67. 

  68. 

  69. cc1 = akg.tvm.expr.Var("cc1", 'int32')

  70. cc2 = akg.tvm.expr.Var("cc2", 'int32')

  71. cc3 = akg.tvm.expr.Var("cc3", 'int32')

  72. cc4 = akg.tvm.expr.Var("cc4", 'int32')

  73. test_comInfo_list = [createComInfo("input_0_local_UB", [1], [32], [cc2], cc3, 'float16'),

  74.                      createComInfo("input_1_local_UB", [1], [32], [cc3], cc3, 'float16'),

  75.                      createComInfo("input_2_local_UB", [32, 1], [16, 32], [cc1, cc2], cc1 * 32 + cc2, 'float16'),

  76.                      createComInfo("input_3_local_UB", [32, 1], [16, 32], [cc1, cc2], cc1 * 32 + cc2, 'float16'),

  77.                      createComInfo("input_4_local_UB", [16, 1], [64, 16], [cc2, cc3], cc2 * 16 + cc3, 'float16'),

  78.                      createComInfo("input_5_local_UB", [512, 32, 1], [16, 16, 32], [cc1, cc2, cc3], cc1 * 512 + cc2 * 32 + cc3, 'float16'),

  79.                      createComInfo("input_6_local_UB", [256, 32, 1], [16, 16, 16], [cc1, cc2, cc3], cc1 * 256 + cc2 * 16 + cc3, 'float16')]

  80. 

  81. 

  82. def testEliminateVarInExpr():

  83.     # expr, elimVars, resultExpr

  84.     test_suite = [[cc3 * 16 + cc4, [cc4], cc3 * 16],

  85.                   [cc1 * 256 + cc2 * 16 + cc3, [cc1, cc3], cc2 * 16],

  86.                   [cc1 * 256 + cc2 * 16 + cc3, [], cc1 * 256 + cc2 * 16 + cc3],

  87.                   [(cc1 * 256 + cc3 * 16) * 32 + cc2 * 32 + cc4, [cc2, cc3], cc1 * 256 * 32 + cc4]]

  88.     idx = 0

  89.     f = akg.tvm.get_global_func("cce_util.EliminateVarInExpr")

  90.     for suite in test_suite:

  91.         result = f(suite[0], suite[1])

  92.         assert str(akg.tvm.ir_pass.Simplify(result)) == str(akg.tvm.ir_pass.Simplify(suite[2])), "case " + str(idx) + " failed"

  93.         idx += 1

  94. 

  95. 

  96. def testGetBufScope():

  97.     test_suite = {'input_1_local_UB': cce.scope_ubuf,

  98.                   'input_1_local_L1': cce.scope_cbuf,

  99.                   'input_1_local_L0A': cce.scope_ca,

  100.                   'input_1_local_L0B': cce.scope_cb,

  101.                   'input_1_local_L0C': cce.scope_cc,

  102.                   'input_1_local_UB_dst_tmp': cce.scope_ubuf,

  103.                   'input_1': cce.dma_copy_global}

  104.     idx = 0

  105.     f = akg.tvm.get_global_func("cce_util.GetBufScope")

  106.     for key, value in test_suite.items():

  107.         result = f(key)

  108.         assert result == value, "case " + str(idx) + " failed"

  109.         idx += 1

  110. 

  111. 

  112. def testGetVarsInExpr():

  113.     test_suite = [[cc3 * 16 + cc4, [cc3, cc4]],

  114.                   [(cc1 * 256 + cc3 * 16) * 32 + cc2 * 32 + cc4, [cc1, cc3, cc2, cc4]],

  115.                   [cc1 * 256 + cc2 * 16 + cc3, [cc1, cc2, cc3]]]

  116.     f = akg.tvm.get_global_func("cce_util.GetVarsInExpr")

  117.     idx = 0

  118.     for suite in test_suite:

  119.         result = f(suite[0])

  120.         assert set(str(result)) == set(str(suite[1])), "case " + str(idx) + " failed"

  121.         idx += 1

  122. 

  123. 

  124. def testIsElementwise():

  125.     test_suite = [[[test_comInfo_list[2]], [test_comInfo_list[2], test_comInfo_list[3]], True],

  126.                   [[test_comInfo_list[0]], [test_comInfo_list[2], test_comInfo_list[0]], False],

  127.                   [[test_comInfo_list[4]], [test_comInfo_list[0], test_comInfo_list[4]], False]]

  128.     f = akg.tvm.get_global_func("cce_util.IsElementwise")

  129.     idx = 0

  130.     for suite in test_suite:

  131.         result = f(suite[0], suite[1])

  132.         assert result == suite[2], "case " + str(idx) + " failed"

  133.         idx += 1

  134. 

  135. 

  136. def testIsBroadcast():

  137.     test_suite = [[[test_comInfo_list[2]], [test_comInfo_list[2], test_comInfo_list[3]], False],

  138.                   [[test_comInfo_list[5]], [test_comInfo_list[4], test_comInfo_list[5]], True],

  139.                   [[test_comInfo_list[4]], [test_comInfo_list[0], test_comInfo_list[4]], True]]

  140.     f = akg.tvm.get_global_func("cce_util.IsBroadcast")

  141.     idx = 0

  142.     for suite in test_suite:

  143.         result = f(suite[0], suite[1])

  144.         assert result == suite[2], "case " + str(idx) + " failed"

  145.         idx += 1

  146. 

  147. 

  148. def testIsLastAxisReduction():

  149.     test_suite = [[[test_comInfo_list[3]], [test_comInfo_list[3], test_comInfo_list[5]], True],

  150.                   [[test_comInfo_list[2]], [test_comInfo_list[2], test_comInfo_list[3]], False],

  151.                   [[test_comInfo_list[4]], [test_comInfo_list[6], test_comInfo_list[4]], False]]

  152.     f = akg.tvm.get_global_func("cce_util.IsLastAxisReduction")

  153.     idx = 0

  154.     for suite in test_suite:

  155.         result = f(suite[0], suite[1])

  156.         assert result == suite[2], "case " + str(idx) + " failed"

  157.         idx += 1

  158. 

  159. 

  160. if __name__ == "__main__":

  161.     testEliminateVarInExpr()

  162.     testGetBufScope()

  163.     testGetVarsInExpr()

  164.     testIsBroadcast()

  165.     testIsElementwise()