Skip to content

Commit db5449b

Browse files
committed
[AMDGPU][ISel] Set trunc store action to expand for v4f32->v4bf16
1 parent 53cda4c commit db5449b

File tree

2 files changed

+431
-0
lines changed

2 files changed

+431
-0
lines changed

llvm/lib/Target/AMDGPU/AMDGPUISelLowering.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -313,11 +313,16 @@ AMDGPUTargetLowering::AMDGPUTargetLowering(const TargetMachine &TM,
313313

314314
setTruncStoreAction(MVT::f32, MVT::bf16, Expand);
315315
setTruncStoreAction(MVT::f32, MVT::f16, Expand);
316+
setTruncStoreAction(MVT::v2f32, MVT::v2bf16, Expand);
316317
setTruncStoreAction(MVT::v2f32, MVT::v2f16, Expand);
317318
setTruncStoreAction(MVT::v3f32, MVT::v3f16, Expand);
319+
setTruncStoreAction(MVT::v4f32, MVT::v4bf16, Expand);
318320
setTruncStoreAction(MVT::v4f32, MVT::v4f16, Expand);
321+
setTruncStoreAction(MVT::v8f32, MVT::v8bf16, Expand);
319322
setTruncStoreAction(MVT::v8f32, MVT::v8f16, Expand);
323+
setTruncStoreAction(MVT::v16f32, MVT::v16bf16, Expand);
320324
setTruncStoreAction(MVT::v16f32, MVT::v16f16, Expand);
325+
setTruncStoreAction(MVT::v32f32, MVT::v32bf16, Expand);
321326
setTruncStoreAction(MVT::v32f32, MVT::v32f16, Expand);
322327

323328
setTruncStoreAction(MVT::f64, MVT::bf16, Expand);

0 commit comments

Comments
 (0)