mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Silence some pytype errors.

Browse Source 
PiperOrigin-RevId: 404122713
This commit is contained in:
Rebecca Chen 2021-10-18 18:46:57 -07:00 committed by jax authors
parent fa89f34be6
commit 6bd0c72f55
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jax/interpreters/xla.py
View File

@ -1344,7 +1344,7 @@ class _DeviceArray(DeviceArray): # type: ignore
def _value(self):
self._check_if_deleted()
if self._npy_value is None:
self._npy_value = self.device_buffer.to_py()
self._npy_value = self.device_buffer.to_py() # pytype: disable=attribute-error # bind-properties
self._npy_value.flags.writeable = False
return self._npy_value
@ -1387,7 +1387,7 @@ class _DeviceArray(DeviceArray): # type: ignore
@property
def __cuda_array_interface__(self):
return self.device_buffer.__cuda_array_interface__
return self.device_buffer.__cuda_array_interface__ # pytype: disable=attribute-error # bind-properties
# Adding methods dynamically to both _DeviceArray and _CppDeviceArray