FBGEMM
FBGEMM copied to clipboard
dense_embedding_codegen_lookup_function CPU is uncorrect with GPU
in version 1.1.0, we test GPU's and CPU's dense_embedding_codegen_lookup_function,found that CPU's output is pooled thought pooling_mode is None. GPU's output is normal. there are some version solve the problem?
Hi @JoheyHan thanks for reporting this issue. We no longer support 1.1.0, as it is superceded by 1.2.0 (and soon to be superceded by 1.3.0). Could you try and see if the issue reproduces on 1.2.0? If so, we will look into this.
Hi @q10 , this issue will still be reproduced in 1.3.0.