MultiScaleDeformableAttention complied but ImportError undefined symbol: THPVariableClass

[RiceZ]

MultiScaleDeformableAttention compiled success：

Installed /home/xxx/anaconda3/envs/env_DINO2/lib/python3.8/site-packages/MultiScaleDeformableAttention-1.0-py3.8-linux-x86_64.egg
Processing dependencies for MultiScaleDeformableAttention==1.0
Finished processing dependencies for MultiScaleDeformableAttention==1.0

when run:
ImportError: /home/zhangchi/anaconda3/envs/env_DINO2/lib/python3.8/site-packages/MultiScaleDeformableAttention-1.0-py3.8-linux-x86_64.egg/MultiScaleDeformableAttention.cpython-38-x86_64-linux-gnu.so: undefined symbol: THPVariableClass

thank you~

[RiceZ]

I got it！

DINO-main\models\dino\ops\modules\ms_deform_attn.py

I comment these codes next before complie MultiScaleDeformableAttention:

    # for amp
    if value.dtype == torch.float16:
        # for mixed precision
        output = MSDeformAttnFunction.apply(
        value.to(torch.float32), input_spatial_shapes, input_level_start_index, sampling_locations.to(torch.float32), attention_weights, self.im2col_step)
        output = output.to(torch.float16)
        output = self.output_proj(output)
        return output

It works, but I donot know why...