DeformableDETR support bfloat16 (#29232)
* Update ms_deform_attn_cuda.cu
* Update ms_deform_attn_cuda.cuh
* Update modeling_deformable_detr.py
* Update src/transformers/models/deformable_detr/modeling_deformable_detr.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
* Update modeling_deformable_detr.py
* python utils/check_copies.py --fix_and_overwrite
* Fix dtype missmatch error
* Update test_modeling_deformable_detr.py
* Update test_modeling_deformable_detr.py
* Update modeling_deformable_detr.py
* Update modeling_deformable_detr.py
* Support DeformableDETR with bfloat16
* Add test code
* Use AT_DISPATCH_FLOATING_TYPES_AND2
Use AT_DISPATCH_FLOATING_TYPES_AND2
* Update tests/models/deformable_detr/test_modeling_deformable_detr.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
* Update tests/models/deformable_detr/test_modeling_deformable_detr.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
* Fix not found require_torch_bf16 function
---------
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
