fix tests when scalar is bfloat16, half
4 files changed