cicdatopea's picture
fix cuda inference code
1b5913e verified