bifrost icon indicating copy to clipboard operation
bifrost copied to clipboard

BFMap does not work with half precision floats

Open dentalfloss1 opened this issue 1 year ago • 1 comments

BFMap fails without good debugging information when trying to use half precision floats. Minimal example:

from bifrost import map as BFMap
import numpy as np
import traceback
for dtype in ['f8','f4','f2','i8','i4','i2']:
    print("Using dtype",dtype)
    try:
        A = BFArray(np.arange(2000,dtype=dtype),space='cuda')
        B = BFArray(np.arange(2000,dtype=dtype),space='cuda')
        C = BFArray(np.arange(2000,dtype=dtype),space='cuda')
        BFMap("""C(i)=A(i)*B(i)""",{'C':C,'B':B,'A':A},axis_names=('i'),shape=(2000,))

        out = C.copy('system')
    except Exception:
        print(traceback.format_exc())```

dentalfloss1 avatar Jun 03 '24 17:06 dentalfloss1

Sounds like it is similar to this: https://github.com/ledatelescope/bifrost/issues/130

jaycedowell avatar Jun 03 '24 17:06 jaycedowell