Browse Source

for pylint 5th

tags/v0.5.0-beta
z00478463 5 years ago
parent
commit
3d3a745a4d
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_left_cast_impl.py
  2. +1
    -1
      mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_right_mul_impl.py

+ 1
- 1
mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_left_cast_impl.py View File

@@ -418,7 +418,7 @@ def cus_cube_matmul_cast(tik_instance, input_x1, trans_a, input_x2, trans_b,
res, mo_tile, ko_tile, no_tile, diag_opt=False, diag_size=128):
"""cus_cube_matmul_cast"""
ko, mo, _, _ = input_x1.shape
no, ko, ki, _ = input_x2.shape
no, ko, _, _ = input_x2.shape
c0 = input_x1.shape[-1]
diag_outer = diag_size // c0
maxblocknum = 32


+ 1
- 1
mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_right_mul_impl.py View File

@@ -99,7 +99,7 @@ def cus_cube_matmul_right_mul(tik_instance, input_x1, input_x2, input_x3,
"""cus_cube_matmul_right_mul"""
diag_size = 128
ko, mo, _, _ = input_x1.shape
no, ko, ki, _ = input_x2.shape
no, ko, _, _ = input_x2.shape
c0 = input_x1.shape[-1]
diag_outer = diag_size // c0
if [input_x1.shape[-1], input_x1.shape[-2], input_x2.shape[-1], input_x2.shape[-2]] != [c0, c0, c0, c0]:


Loading…
Cancel
Save