Implement `__enter__` and `__exit__` for CUDA context

[riaqn]

People like me always forgot to pop context and cause errors. By implementing these two methods, the python will handle the context automatically. See https://book.pythontips.com/en/latest/context_managers.html This also helps when there is an exception.

[inducer]

That sounds reasonable. I'd be happy to take a PR.