I've got the same issue. For what its worth, casting to float16 works as well. The issue also only occurs when the tensor is on the GPU. If you move the tensor to the CPU, then sort, sorting works as expected.
be2112
Users receive reputation points by contributing quality content to the forums.
Posts include post and replies published on the forums.
Post authors can mark replies as Accepted to indicate successful solutions.
Apple Developer Forums admins can mark replies as Apple Recommended to indicate an approved solution
Last seen
User for