RuntimeError: Compile graph kernel_graph0 failed.

import mindspore as ms
from mindspore import mint, Tensor, value_and_grad
import numpy as np

ms.set_context(mode=ms.GRAPH_MODE)
input_data = np.array([0, 1])
other_data = np.array([1, 1])
ms_input = Tensor(input_data, ms.float32)
ms_other = Tensor(other_data, ms.float32)

def forward_ms(x, y):
    return mint.arctan2(x, y)

# Gradients for MindSpore
grad_fn = value_and_grad(forward_ms)
output_ms, gradient_ms = grad_fn(ms_input, ms_other)

# 打印输出和梯度
print("MindSpore output:", output_ms)
print("MindSpore gradient:", gradient_ms)