valhalla's picture
add fp16 version
1ea4f60