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.
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
MindSpore-AKG/python/akg/composite/repository_gpu.json

73 lines
2.5 kB
Raw Permalink Blame History 

  1. {

  2.     "2.Mul11.Mul1.TensorAdd1.Mul1.Mul45.Mul1.TensorAdd17.Mul1.Tanh1.Mul11.Sub1.Mul12.Mul12.Mul110.TensorAdd6.Mul1.TensorAdd13.Mul119.19": {

  3.         "8192_3072--": {

  4.             "float16--": {

  5.                 "dim": "0 0 1024 1024",

  6.                 "bind_block": "12288",

  7.                 "bind_thread": "1024"

  8.             }

  9.         }

  10.     },

  11.     "2.Cast1.TensorAdd13.Mul11.Mul12.Mul1.TensorAdd14.Mul1.Tanh1.TensorAdd1.Mul18.Mul1.312": {

  12.         "metadata": {

  13.             "attrs": {

  14.                 "enable_auto_fuse": false

  15.             }

  16.         },

  17.         "8192_3072.3072.8192_3072-": {

  18.             "float16.float32.float16-": {

  19.                 "dim": "0 0 4 4 0 1 1024 1024",

  20.                 "bind_block": "3 1024 1",

  21.                 "bind_thread": "1024 1 1"

  22.             }

  23.         }

  24.     },

  25.     "3.Mul2.Mul14.Mul13.ReduceSum1.46": {

  26.         "64_12_128_128---.64_12_128_1": {

  27.             "float16----": {

  28.                 "dim": "0 0 128 128 0 1 128 128",

  29.                 "bind_block": "1 768 1",

  30.                 "bind_thread": "32 32 1"

  31.             }

  32.         }

  33.     },

  34.     "2.Cast2.TensorAdd12.Reshape1.Transpose1.5": {

  35.         "768.8192_768.64_12_128_64": {

  36.             "float32.float16-": {

  37.                 "dim": "0 0 1 1 0 1 1 1 0 2 16 16 0 3 64 64",

  38.                 "bind_block": "1 12 64",

  39.                 "bind_thread": "64 16 1"

  40.             }

  41.         }

  42.     },

  43.     "2.BatchMatMul12.2": {

  44.         "any_shape": {

  45.             "float16--": {

  46.                 "metadata": {

  47.                     "attrs": {

  48.                         "use_shared_memory": true,

  49.                         "pragma_enable_tensor_core": true,

  50.                         "pragma_enable_matmul": true,

  51.                         "enable_auto_fuse": false,

  52.                         "vector_load_type": "float4",

  53.                         "bind_thread": "32 4"

  54.                     }

  55.                 }

  56.             }

  57.         }

  58.     },

  59.     "12.GreaterEqual6.Cast1.LessEqual7.Cast1.AddN8.Reshape1.Cast1.Mul14.Mul17.Mul1.Mul117.Mul121.Sub2021.Mul1.Mul225.Mul14.Mul1.ReduceSum1.Mul730.Mul731.Mul110.22232425293032": {

  60.         "metadata": {

  61.             "attrs": {

  62.                 "enable_atomic_add": true

  63.             }

  64.         },

  65.         "64_128_1-.768.64_128_768.64_128_1.64_128_768--.8192_768---.64_128_768-.64_128_1.64_128_768---": {

  66.             "float32-------.float16---.float32------": {

  67.                 "dim": "0 0 8 1 0 1 768 1",

  68.                 "bind_block": "1 1024",

  69.                 "bind_thread": "128 8"

  70.             }

  71.         }

  72.     }

  73. }