Skip to content

Commit 94749db

Browse files
pkowalc1rkazants
andauthoredSep 22, 2024
[TF][GPU]: Enable bitwise shift test in TF. (openvinotoolkit#26194)
### Details: - Enabled TF tests for Bitwise shift ops on GPU This PR depends on openvinotoolkit#26251, which adds support for int16 and uint16 data types in elementwise shift op in gpu plugin. Also depends on openvinotoolkit#26383, which fixes a bug related to i16 support for elementwise ops in GPU. --------- Co-authored-by: Roman Kazantsev <roman.kazantsev@intel.com>
1 parent 538374f commit 94749db

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed
 

‎tests/layer_tests/tensorflow_tests/test_tf_BitwiseShift.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,8 @@ def create_bitwise_net(self, x_shape, y_shape, is_y_const, input_type, op_type):
6868
@pytest.mark.nightly
6969
def test_bitwise(self, x_shape, y_shape, is_y_const, input_type, op_type, ie_device, precision, ir_version,
7070
temp_dir, use_legacy_frontend):
71-
if ie_device == 'GPU':
72-
pytest.skip("149424: Bitwise ops are not supported on GPU")
71+
if ie_device == 'GPU' and input_type in [np.uint64]:
72+
pytest.skip("149424: uint64 type is not supported on GPU")
7373
if use_legacy_frontend:
7474
pytest.skip("BitwiseShift ops are supported only by new TF FE")
7575
self._test(*self.create_bitwise_net(x_shape=x_shape, y_shape=y_shape, is_y_const=is_y_const,

0 commit comments

Comments
 (0)
Please sign in to comment.