/*********************************************************************/

/* Copyright 2009, 2010 The University of Texas at Austin.           */

/* All rights reserved.                                              */

/*                                                                   */

/* Redistribution and use in source and binary forms, with or        */

/* without modification, are permitted provided that the following   */

/* conditions are met:                                               */

/*                                                                   */

/*   1. Redistributions of source code must retain the above         */

/*      copyright notice, this list of conditions and the following  */

/*      disclaimer.                                                  */

/*                                                                   */

/*   2. Redistributions in binary form must reproduce the above      */

/*      copyright notice, this list of conditions and the following  */

/*      disclaimer in the documentation and/or other materials       */

/*      provided with the distribution.                              */

/*                                                                   */

/*    THIS  SOFTWARE IS PROVIDED  BY THE  UNIVERSITY OF  TEXAS AT    */

/*    AUSTIN  ``AS IS''  AND ANY  EXPRESS OR  IMPLIED WARRANTIES,    */

/*    INCLUDING, BUT  NOT LIMITED  TO, THE IMPLIED  WARRANTIES OF    */

/*    MERCHANTABILITY  AND FITNESS FOR  A PARTICULAR  PURPOSE ARE    */

/*    DISCLAIMED.  IN  NO EVENT SHALL THE UNIVERSITY  OF TEXAS AT    */

/*    AUSTIN OR CONTRIBUTORS BE  LIABLE FOR ANY DIRECT, INDIRECT,    */

/*    INCIDENTAL,  SPECIAL, EXEMPLARY,  OR  CONSEQUENTIAL DAMAGES    */

/*    (INCLUDING, BUT  NOT LIMITED TO,  PROCUREMENT OF SUBSTITUTE    */

/*    GOODS  OR  SERVICES; LOSS  OF  USE,  DATA,  OR PROFITS;  OR    */

/*    BUSINESS INTERRUPTION) HOWEVER CAUSED  AND ON ANY THEORY OF    */

/*    LIABILITY, WHETHER  IN CONTRACT, STRICT  LIABILITY, OR TORT    */

/*    (INCLUDING NEGLIGENCE OR OTHERWISE)  ARISING IN ANY WAY OUT    */

/*    OF  THE  USE OF  THIS  SOFTWARE,  EVEN  IF ADVISED  OF  THE    */

/*    POSSIBILITY OF SUCH DAMAGE.                                    */

/*                                                                   */

/* The views and conclusions contained in the software and           */

/* documentation are those of the authors and should not be          */

/* interpreted as representing official policies, either expressed   */

/* or implied, of The University of Texas at Austin.                 */

/*********************************************************************/

#define ASSEMBLER

#include "common.h"

#ifdef XDOUBLE

#define PREFETCH_SIZE ( 8 * 16)

#elif defined(DOUBLE)

#define PREFETCH_SIZE (16 * 16)

#else

#define PREFETCH_SIZE (32 * 16)

#endif

#ifndef COMPLEX

#define COMPADD	0

#define STRIDE INCX

#else

#define COMPADD	1

#define STRIDE SIZE

#endif

#define PRE1	r2

#define I	r17

#define J	r18

#define X2	r19

#define INCX5	r20

#define INCX16	r21

#define N	r32

#define X	r33

#define INCX	r34

#define PR	r30

#define ARLC	r31

	PROLOGUE

	.prologue

	PROFCODE

	{ .mfi

	adds	PRE1 = PREFETCH_SIZE * SIZE, X

	mov	f8  = f0

	.save ar.lc, ARLC

	mov	ARLC = ar.lc

	}

	;;

	.body

#ifdef F_INTERFACE

	LDINT	N    = [N]

	LDINT	INCX = [INCX]

	;;

#ifndef USE64BITINT

	sxt4	N = N

	sxt4	INCX = INCX

	;;

#endif

#endif

	{ .mmi

	cmp.ge	p6, p0 = r0, N

	cmp.ge	p7, p0 = r0, INCX

	shr	I =  N, (4 - COMPADD)

	}

	{ .mbb

	and	J = ((1 << (4 - COMPADD)) - 1), N

	(p6) 	br.ret.sptk.many b0

	(p7) 	br.ret.sptk.many b0

	}

	;;

	{ .mfi

	mov	f9  = f0

	mov	PR = pr

	}

	{ .mfi

	adds	I = -1, I

	mov	f10 = f0

	shl	INCX = INCX, (BASE_SHIFT + COMPADD)

	}

	;;

	{ .mfi

	shladd	X2     = INCX,  (2 - COMPADD), X

	mov	f11 = f0

	mov	pr.rot = 0

	}

	{ .mfi

	shladd	INCX5  = INCX,  (2 - COMPADD), INCX

	mov	f12 = f0

	tbit.z	p0, p12 = N, (3 - COMPADD)

	}

	;;

	{ .mfi

	shladd	INCX16  = INCX, (4 - COMPADD), r0

	mov	f13 = f0

	mov	ar.ec= 3

	}

	{ .mmf

	cmp.gt	p8 ,p0  =   r0, I

	cmp.eq	p16, p0 = r0, r0

	mov	f14 = f0

	}

	;;

	{ .mmf

#ifdef COMPLEX

       adds	INCX  = - SIZE, INCX

       adds	INCX5 = - SIZE, INCX5

#else

	nop.m	0

	nop.m	0

#endif

	mov	f15 = f0

	}

	{ .mib

	cmp.eq	p9, p0  =   r0, J

	mov	ar.lc = I

	(p8) br.cond.dpnt  .L52

	}

	;;

	.align 32

.L51:

	(p16) LDFD	f32  = [X],  STRIDE	

	(p16) lfetch.nt1 [PRE1], INCX16

	(p18) fma.d.s1	f8  = f34, f34, f8

	(p16) LDFD	f35  = [X2], STRIDE

	(p18) fma.d.s1	f9  = f37, f37, f9

	nop.b 0

	;;

	(p16) LDFD	f38  = [X],  INCX

	(p18) fma.d.s1	f10 = f40, f40, f10

	nop.b 0

	(p16) LDFD	f41  = [X2], INCX

	(p18) fma.d.s1	f11 = f43, f43, f11

	nop.b 0

	;;

	(p16) LDFD	f44  = [X],  STRIDE

	(p18) fma.d.s1	f12 = f46, f46, f12

	nop.b 0

	(p16) LDFD	f47  = [X2], STRIDE

	(p18) fma.d.s1	f13 = f49, f49, f13

	nop.b 0

	;;

	(p16) LDFD	f50  = [X],  INCX5

	(p18) fma.d.s1	f14 = f52, f52, f14

	nop.b 0

	(p16) LDFD	f53  = [X2], INCX5

	(p18) fma.d.s1	f15 = f55, f55, f15

	nop.b 0

	;;

	(p16) LDFD	f56  = [X],  STRIDE

	(p18) fma.d.s1	f8  = f58, f58, f8

	nop.b 0

	(p16) LDFD	f59  = [X2], STRIDE

	(p18) fma.d.s1	f9  = f61, f61, f9

	nop.b 0

	;;

	(p16) LDFD	f62  = [X],  INCX

	(p18) fma.d.s1	f10 = f64, f64, f10

	nop.b 0

	(p16) LDFD	f65  = [X2], INCX

	(p18) fma.d.s1	f11 = f67, f67, f11

	nop.b 0

	;;

	(p16) LDFD	f68  = [X],  STRIDE

	(p18) fma.d.s1	f12 = f70, f70, f12

	nop.b 0

	(p16) LDFD	f71  = [X2], STRIDE	

	(p18) fma.d.s1	f13 = f73, f73, f13

	nop.b 0

	;;

	(p16) LDFD	f74  = [X],  INCX5

	(p18) fma.d.s1	f14 = f76, f76, f14

	nop.b 0

	(p16) LDFD	f77  = [X2], INCX5

	(p18) fma.d.s1	f15 = f79, f79, f15

	br.ctop.sptk.few .L51

	;;

	.align 32

.L52:

	{ .mmb

	(p12) LDFD	f32  = [X],  STRIDE

	(p12) LDFD	f33  = [X2], STRIDE

	(p9) br.cond.dptk .L998

	}

	;;

	{ .mmi

	(p12) LDFD	f34  = [X],  INCX

	(p12) LDFD	f35  = [X2], INCX

	tbit.z	p0, p13 = N, (2 - COMPADD)

	}

	;;

	{ .mmi

	(p12) LDFD	f36  = [X],  STRIDE

	(p12) LDFD	f37  = [X2], STRIDE

	tbit.z	p0, p14 = N, (1 - COMPADD)

	}

	;;

	{ .mmi

	(p12) LDFD	f38  = [X],  INCX5

	(p12) LDFD	f39  = [X2], INCX5

#ifndef COMPLEX

	tbit.z	p0, p15 = N, 0

#endif

	}

	;;

	(p13) LDFD	f40  = [X], STRIDE

	(p12) fma.d.s1	f8  = f32, f32, f8

	(p12) fma.d.s1	f9  = f33, f33, f9

	;;

	(p13) LDFD	f41  = [X], INCX

	(p12) fma.d.s1	f10 = f34, f34, f10

	(p12) fma.d.s1	f11 = f35, f35, f11

	;;

	(p13) LDFD	f42  = [X], STRIDE

	(p12) fma.d.s1	f12 = f36, f36, f12

	(p12) fma.d.s1	f13 = f37, f37, f13

	;;

	(p13) LDFD	f43  = [X], INCX

	(p12) fma.d.s1	f14 = f38, f38, f14

	(p12) fma.d.s1	f15 = f39, f39, f15

	;;

	(p14) LDFD	f44  = [X], STRIDE

	(p13) fma.d.s1	f8  = f40, f40, f8

	(p13) fma.d.s1	f9  = f41, f41, f9

	;;

	(p14) LDFD	f45  = [X], INCX

	(p13) fma.d.s1	f10 = f42, f42, f10

	(p13) fma.d.s1	f11 = f43, f43, f11

	;;

#ifndef COMPLEX

	(p15) LDFD	f46  = [X]

#endif

	(p14) fma.d.s1	f12 = f44, f44, f12

	(p14) fma.d.s1	f13 = f45, f45, f13

	;;

#ifndef COMPLEX

	(p15) fma.d.s1	f14 = f46, f46, f14

	;;		      

#endif

	.align 32

.L998:

	{ .mmf

	fadd.d.s1	f8  = f8,  f9

	}

	{ .mmf

	fadd.d.s1	f10 = f10, f11

	}

	{ .mmf

	fadd.d.s1	f12 = f12, f13

	}

	{ .mfi

	fadd.d.s1	f14 = f14, f15

	mov	ar.lc = ARLC

	}

	;;

	{ .mmf

	fadd.d.s1	f8  = f8,  f10

	}

	{ .mfi

	fadd.d.s1	f12 = f12, f14

	mov	pr = PR, -65474

	}

	;;

	{ .mfb

	fadd.d.s1	f8  = f8,  f12

 	br	sqrt

	}

	;;

	EPILOGUE

	.section .data

	.type	sqrt, @function

	.global	sqrt