rocm

I see comments suggesting adding this to understand how loops are being handled by numba, and in the their own FAQ (https://numba.pydata.org/numba-doc/latest/user/faq.html)

from llvmlite import binding as llvm
llvm.set_option('','--debug-only=loop-vectorize')

You would then create your njit function and run it, and I believe the idea is that it prints debug information about whether

Description

https://numpy.org/doc/stable/reference/generated/numpy.corrcoef.html

https://docs.cupy.dev/en/stable/reference/generated/cupy.corrcoef.html

Seems args are different

Additional Information

dtype argument added in NumPy version 1.20.

Please go to https://github.com/deepmodeling/deepmd-kit/discussions to ask questions. Before asking questions, you can

• search the previous discussions
• check the document, especially training parameters

Please provide necessary information including the version of software and installation way,

Just an FYI whilst I was trawling through the ROCm GitHub page:

https://rocmdocs.amd.com/en/latest/Programming_Guides/Programming-Guides.html#

Since arrays may not actually be modified by a given operation, or might only be partially modified (or the user has some other way to ensure correctness).