# preparation
Accordding to the following PR to produce fmhalib.so
https://github.com/sneaxiy/apex/pull/1

# build the fmha op
python setup.py install

# unittest (need to improve)
python test_fmha.py