NumPy-CuPy Generic Code SupportΒΆ

cupy.get_array_module Returns the array module for arguments.