tests/test_fastnn/test_basic_ops.py · ea8bfcb0a396f27f2a74d74a87f3084a8f6dd400 · OpenDAS / FastFold

add unit test for layernorm, linear, outproductmean (#82) · ea8bfcb0

oahzxl authored Oct 12, 2022

* update out product mean test

* polish out product mean test and update linear test

* update test for layernorm, use both kernel

* support test without triton

* polish layernorm

* use current code

ea8bfcb0

test_basic_ops.py 542 Bytes

Replace test_basic_ops.py