Update gpu.py, fix argument count