Fix OpCheckError

✅ Solution

OpCheckError usually arises when a custom operator's forward and backward implementations produce results with incorrect data types, shapes, or values compared to the expected output based on the OpInfo specification. To fix it, carefully review the custom operator's forward and backward functions, ensuring that the output tensor's data type, shape, and values match the expected behavior defined in the OpInfo, and use `torch.testing.assert_close` for numerical comparisons in your test.