ptx/src/test/spirv_run/set_f16x2.ptx


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

.version 6.5
.target sm_53
.address_size 64

.visible .entry set_f16x2(
	.param .u64 input,
	.param .u64 output
)
{
	.reg .u64 	    in_addr;
    .reg .u64 	    out_addr;
    .reg .b32 	    temp0;
    .reg .b32 	    temp1;
    .reg .b32 	    temp2;
    .reg .b32 	    temp3;
    .reg .f16x2     sela;

	ld.param.u64 	in_addr, [input];
    ld.param.u64 	out_addr, [output];

    ld.u32              temp0, [in_addr];
    ld.u32              temp1, [in_addr+4];
    ld.u32              temp2, [in_addr+8];
    ld.u32              temp3, [in_addr+12];
    set.gtu.u32.f16x2   temp0, temp0, temp1;
    set.eq.f16x2.f16x2  temp2, temp2, temp3;
    st.b32              [out_addr], temp0;
    st.b32              [out_addr+4], temp2;
	ret;
}