Add test case.
Add flag to disable cuda.