|
- //
- // Generated by NVIDIA NVVM Compiler
- //
- // Compiler Build ID: CL-26907403
- // Cuda compilation tools, release 10.1, V10.1.243
- // Based on LLVM 3.4svn
- //
-
- .version 6.4
- .target sm_60
- .address_size 64
-
- // .globl Fused_ReduceSum_split_214557381024138315_kernel0
- // _ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E18input_0_red_shared has been demoted
- // _ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E8red_buf0 has been demoted
-
- .visible .entry Fused_ReduceSum_split_214557381024138315_kernel0(
- .param .u64 Fused_ReduceSum_split_214557381024138315_kernel0_param_0,
- .param .u64 Fused_ReduceSum_split_214557381024138315_kernel0_param_1
- )
- {
- .reg .pred %p<17>;
- .reg .f32 %f<42>;
- .reg .b32 %r<35>;
- .reg .b64 %rd<9>;
- // demoted variable
- .shared .align 4 .b8 _ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E18input_0_red_shared[4];
- // demoted variable
- .shared .align 4 .b8 _ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E8red_buf0[2048];
-
- ld.param.u64 %rd3, [Fused_ReduceSum_split_214557381024138315_kernel0_param_0];
- ld.param.u64 %rd4, [Fused_ReduceSum_split_214557381024138315_kernel0_param_1];
- mov.u32 %r1, %tid.x;
- setp.ne.s32 %p3, %r1, 0;
- @%p3 bra BB0_2;
-
- mov.u32 %r6, 0;
- st.shared.u32 [_ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E18input_0_red_shared], %r6;
-
- BB0_2:
- cvta.to.global.u64 %rd1, %rd3;
- bar.sync 0;
- mov.u32 %r2, %ctaid.y;
- mad.lo.s32 %r7, %r2, 961, %r1;
- mul.wide.s32 %rd5, %r7, 4;
- add.s64 %rd2, %rd1, %rd5;
- mov.f32 %f39, 0f00000000;
- setp.gt.s32 %p4, %r1, 960;
- mov.f32 %f40, %f39;
- @%p4 bra BB0_4;
-
- ld.global.nc.f32 %f10, [%rd2];
- add.f32 %f39, %f10, 0f00000000;
- sub.f32 %f40, %f39, %f10;
-
- BB0_4:
- add.s32 %r8, %r1, 512;
- setp.gt.s32 %p5, %r8, 960;
- @%p5 bra BB0_6;
-
- ld.global.nc.f32 %f11, [%rd2+2048];
- sub.f32 %f12, %f11, %f40;
- add.f32 %f39, %f39, %f12;
-
- BB0_6:
- mov.u32 %r9, %tid.y;
- mov.u32 %r10, %ntid.x;
- mad.lo.s32 %r11, %r9, %r10, %r1;
- and.b32 %r3, %r11, 511;
- and.b32 %r4, %r11, -512;
- add.s32 %r12, %r4, %r3;
- shl.b32 %r13, %r12, 2;
- mov.u32 %r14, _ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E8red_buf0;
- add.s32 %r5, %r14, %r13;
- st.shared.f32 [%r5], %f39;
- bar.sync 0;
- setp.gt.u32 %p6, %r3, 255;
- @%p6 bra BB0_8;
-
- ld.shared.f32 %f13, [%r5];
- ld.shared.f32 %f14, [%r5+1024];
- add.f32 %f15, %f13, %f14;
- st.shared.f32 [%r5], %f15;
-
- BB0_8:
- bar.sync 0;
- setp.gt.u32 %p7, %r3, 127;
- @%p7 bra BB0_10;
-
- ld.shared.f32 %f16, [%r5];
- ld.shared.f32 %f17, [%r5+512];
- add.f32 %f18, %f16, %f17;
- st.shared.f32 [%r5], %f18;
-
- BB0_10:
- bar.sync 0;
- setp.gt.u32 %p8, %r3, 63;
- @%p8 bra BB0_12;
-
- ld.shared.f32 %f19, [%r5];
- ld.shared.f32 %f20, [%r5+256];
- add.f32 %f21, %f19, %f20;
- st.shared.f32 [%r5], %f21;
-
- BB0_12:
- bar.sync 0;
- setp.gt.u32 %p9, %r3, 31;
- @%p9 bra BB0_14;
-
- ld.shared.f32 %f22, [%r5];
- ld.shared.f32 %f23, [%r5+128];
- add.f32 %f24, %f22, %f23;
- st.shared.f32 [%r5], %f24;
-
- BB0_14:
- setp.lt.u32 %p1, %r3, 32;
- bar.sync 0;
- @!%p1 bra BB0_17;
- bra.uni BB0_15;
-
- BB0_15:
- ld.shared.f32 %f25, [%r5];
- mov.b32 %r15, %f25;
- mov.u32 %r16, 2;
- mov.u32 %r17, 31;
- mov.u32 %r18, 16;
- mov.u32 %r19, -1;
- shfl.sync.down.b32 %r20|%p10, %r15, %r18, %r17, %r19;
- mov.b32 %f26, %r20;
- add.f32 %f27, %f25, %f26;
- mov.b32 %r21, %f27;
- mov.u32 %r22, 8;
- shfl.sync.down.b32 %r23|%p11, %r21, %r22, %r17, %r19;
- mov.b32 %f28, %r23;
- add.f32 %f29, %f27, %f28;
- mov.b32 %r24, %f29;
- mov.u32 %r25, 4;
- shfl.sync.down.b32 %r26|%p12, %r24, %r25, %r17, %r19;
- mov.b32 %f30, %r26;
- add.f32 %f31, %f29, %f30;
- mov.b32 %r27, %f31;
- shfl.sync.down.b32 %r28|%p13, %r27, %r16, %r17, %r19;
- mov.b32 %f32, %r28;
- add.f32 %f33, %f31, %f32;
- mov.b32 %r29, %f33;
- mov.u32 %r30, 1;
- shfl.sync.down.b32 %r31|%p14, %r29, %r30, %r17, %r19;
- mov.b32 %f34, %r31;
- add.f32 %f7, %f33, %f34;
- setp.ne.s32 %p15, %r3, 0;
- @%p15 bra BB0_17;
-
- st.shared.f32 [%r5], %f7;
-
- BB0_17:
- bar.sync 0;
- setp.ne.s32 %p16, %r3, 0;
- @%p16 bra BB0_19;
-
- ld.shared.f32 %f35, [_ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E18input_0_red_shared];
- shl.b32 %r32, %r4, 2;
- add.s32 %r34, %r14, %r32;
- ld.shared.f32 %f36, [%r34];
- add.f32 %f37, %f35, %f36;
- st.shared.f32 [_ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E18input_0_red_shared], %f37;
-
- BB0_19:
- setp.eq.s32 %p2, %r1, 0;
- bar.sync 0;
- @!%p2 bra BB0_21;
- bra.uni BB0_20;
-
- BB0_20:
- ld.shared.f32 %f38, [_ZZ48Fused_ReduceSum_split_214557381024138315_kernel0E18input_0_red_shared];
- cvta.to.global.u64 %rd6, %rd4;
- mul.wide.s32 %rd7, %r2, 4;
- add.s64 %rd8, %rd6, %rd7;
- st.global.f32 [%rd8], %f38;
-
- BB0_21:
- bar.sync 0;
- ret;
- }
-
|