Skip to content

maki49/AnTrans-RI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AnTrans-RI

This code is a Tensor-Transpose Analyzer for LibRI.

( also, it represents my Entrance into the underlying algorithm of LibRI. )

Given the labels of a list of 2-or-3-dimentional tensors to be multiplied, this code will search an optimized sequence of following operations:

  • tensor3_transpose
  • tensor3_merge
  • gemm('NN'/'TN'/'NT'/'NN')
  • exchange

to do the multiplication correctly, which minimizes the number of calls to the time-consuming functiontensor3_transpose.


Usage

make with googletest and run ./search_test > result.txt.


Results

cal_func_b01s in LibRI have been written as test cases.

cal_func_bx2s in LibRI are input to the search algorithm.


Actually, I don't think this code is really useful, you know ...

I just regard it as a game, which reminds me of my undergraduate years.

About

a tensor-transpose analyzer for LibRI

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published