pystencils.sympyextensions.fast_approximation.fast_sqrt

Contents

pystencils.sympyextensions.fast_approximation.fast_sqrt#

class pystencils.sympyextensions.fast_approximation.fast_sqrt(*args)#

Perform a fast approximate square root on GPU targets.

This function is only supported on CUDA and HIP targets, where it will resolve to __fsqrt_rn. Using it on other targets will lead to an error during code generation.