tf.raw_ops.ResourceScatterDiv crash with abortion

[trickiwoo]

Click to expand!

Issue Type

Bug

Have you reproduced the bug with TF nightly?

Yes

Source

source

Tensorflow Version

2.13.0-dev20230208

Custom Code

Yes

OS Platform and Distribution

No response

Mobile device

No response

Python version

No response

Bazel version

No response

GCC/Compiler version

No response

CUDA/cuDNN version

8.2.4

GPU model and memory

No response

Current Behaviour?

2.13.0-dev20230208

Standalone code to reproduce the issue

import tensorflow as tf
from tensorflow.python.eager import context

input1 = tf.raw_ops.VarHandleOp(dtype=tf.int32, shape=[2, 3], shared_name=context.anonymous_name())
input2 = tf.constant([],dtype=tf.float32)
output = tf.raw_ops.ResourceScatterDiv(resource=input1, indices=[0], updates=input2)

Relevant log output

2023-03-26 22:11:23.135344: F tensorflow/core/framework/tensor.cc:770] Check failed: dtype() == expected_dtype (3 vs. 1) float expected, got int32
Aborted (core dumped)

[synandi]

Hi @trickiwoo
Thank you for reporting the issue!
I was able to replicate the issue using the latest tf-nightly(2.13.0.dev20230327). Please find the screenshot below.

We are investigating this issue and will update here soon. Thank you!

[tilakrayal]

@trickiwoo,
There was a PR that was raised internally and the developer is working on that issue. Once it is resolved we will update the result here. Thank you!

[tilakrayal]

@trickiwoo,
I tried to execute the mentioned code on tf-nightly and the code was executed with the error and also observed that the crash did not happen. And the same has been in the respective files. Kindly find the gist for the reference.

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/quantize_and_dequantize_op.cc#L22

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/resource_variable_ops.cc#L1120

// Check data type of update and resource to scatter.
const DataType update_dtype = c->input(2).dtype();
OP_REQUIRES(c, v->tensor()->dtype() == update_dtype,
errors::InvalidArgument(
"DType of scatter resource and updates does not match."));

Thank you!

[github-actions]

This issue is stale because it has been open for 7 days with no activity. It will be closed if no further activity occurs. Thank you.

[github-actions]

This issue was closed because it has been inactive for 7 days since being marked as stale. Please reopen if you'd like to work on this further.

[google-ml-butler]

Are you satisfied with the resolution of your issue?
Yes
No