You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Bob Zhu 220eef602e Extend Matmul to support matrix multiplication with multiple heads (#18570) ... * extend matmul op to support multiple head multiplication With the support of multiple head, the multiplication of two big matrixes is split into multiplication of several (head_number) small matrixes. e.g. if Mat A is [3, 24] and Mat B is [24, 4], when multiple A and B with head_number as 4, Mat A will be split as 4 matrix of [3, 6] and Mat B will be 4 matrix of [6, 4]. The result of final matrix will be 4 matrix of [3, 4], i.e. [3, 16].		6 years ago
..
contrib/float16	fix deadlink (#16129)	6 years ago
fluid	Extend Matmul to support matrix multiplication with multiple heads (#18570)	6 years ago
scripts	Enhance backward process (#18700)	6 years ago
testing	Add flags for init and re-alloc gpu	6 years ago
.common_test_util.sh	fix spelling errors (#17941)	6 years ago
.gitignore	…
.set_port.sh	…
.set_python_path.sh	…
CMakeLists.txt	update CMakeLists.txt	6 years ago