nb_kernel_x86_64_sse.c

来自「最著名最快的分子模拟软件」· C语言 代码 · 共 267 行

/* -*- mode: c; tab-width: 4; indent-tabs-mode: n; c-basic-offset: 4 -*- 
 *
 * $Id: nb_kernel_x86_64_sse.c,v 1.1 2004/12/26 19:33:01 lindahl Exp $
 * 
 * This file is part of Gromacs        Copyright (c) 1991-2004
 * David van der Spoel, Erik Lindahl, University of Groningen.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * To help us fund GROMACS development, we humbly ask that you cite
 * the research papers on the package. Check out http://www.gromacs.org
 * 
 * And Hey:
 * Gnomes, ROck Monsters And Chili Sauce
 */
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

/* Must come directly after config.h */
#include <gmx_thread.h>

#include <types/simple.h>
#include <types/nrnb.h>

#include "nb_kernel_x86_64_sse.h"

/* Include x86_64 SSE kernel headers in local directory */
#include "nb_kernel010_x86_64_sse.h"
#include "nb_kernel030_x86_64_sse.h"
#include "nb_kernel100_x86_64_sse.h"
#include "nb_kernel101_x86_64_sse.h"
#include "nb_kernel102_x86_64_sse.h"
#include "nb_kernel103_x86_64_sse.h"
#include "nb_kernel104_x86_64_sse.h"
#include "nb_kernel110_x86_64_sse.h"
#include "nb_kernel111_x86_64_sse.h"
#include "nb_kernel112_x86_64_sse.h"
#include "nb_kernel113_x86_64_sse.h"
#include "nb_kernel114_x86_64_sse.h"
#include "nb_kernel130_x86_64_sse.h"
#include "nb_kernel131_x86_64_sse.h"
#include "nb_kernel132_x86_64_sse.h"
#include "nb_kernel133_x86_64_sse.h"
#include "nb_kernel134_x86_64_sse.h"
#include "nb_kernel200_x86_64_sse.h"
#include "nb_kernel201_x86_64_sse.h"
#include "nb_kernel202_x86_64_sse.h"
#include "nb_kernel203_x86_64_sse.h"
#include "nb_kernel204_x86_64_sse.h"
#include "nb_kernel210_x86_64_sse.h"
#include "nb_kernel211_x86_64_sse.h"
#include "nb_kernel212_x86_64_sse.h"
#include "nb_kernel213_x86_64_sse.h"
#include "nb_kernel214_x86_64_sse.h"
#include "nb_kernel230_x86_64_sse.h"
#include "nb_kernel231_x86_64_sse.h"
#include "nb_kernel232_x86_64_sse.h"
#include "nb_kernel233_x86_64_sse.h"
#include "nb_kernel234_x86_64_sse.h"
#include "nb_kernel300_x86_64_sse.h"
#include "nb_kernel301_x86_64_sse.h"
#include "nb_kernel302_x86_64_sse.h"
#include "nb_kernel303_x86_64_sse.h"
#include "nb_kernel304_x86_64_sse.h"
#include "nb_kernel310_x86_64_sse.h"
#include "nb_kernel311_x86_64_sse.h"
#include "nb_kernel312_x86_64_sse.h"
#include "nb_kernel313_x86_64_sse.h"
#include "nb_kernel314_x86_64_sse.h"
#include "nb_kernel330_x86_64_sse.h"
#include "nb_kernel331_x86_64_sse.h"
#include "nb_kernel332_x86_64_sse.h"
#include "nb_kernel333_x86_64_sse.h"
#include "nb_kernel334_x86_64_sse.h"
#include "nb_kernel400_x86_64_sse.h"
#include "nb_kernel410_x86_64_sse.h"
#include "nb_kernel430_x86_64_sse.h"




#include <stdlib.h>
#include <stdio.h>
/* Necessary headers for POSIX-style long jumps. */
#include <signal.h>
#include <setjmp.h>



#include "../nb_kerneltype.h"
#include "nb_kernel_x86_64_sse.h"
#include "nb_kernel_x86_64_sse_test_asm.h"


static nb_kernel_t *
kernellist_x86_64_sse[eNR_NBKERNEL_NR] = 
{
    nb_kernel010_x86_64_sse,
    NULL,
    nb_kernel030_x86_64_sse,
    nb_kernel100_x86_64_sse,
    nb_kernel101_x86_64_sse,
    nb_kernel102_x86_64_sse,
    nb_kernel103_x86_64_sse,
    nb_kernel104_x86_64_sse,
    nb_kernel110_x86_64_sse,
    nb_kernel111_x86_64_sse,
    nb_kernel112_x86_64_sse,
    nb_kernel113_x86_64_sse,
    nb_kernel114_x86_64_sse,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    nb_kernel130_x86_64_sse,
    nb_kernel131_x86_64_sse,
    nb_kernel132_x86_64_sse,
    nb_kernel133_x86_64_sse,
    nb_kernel134_x86_64_sse,
    nb_kernel200_x86_64_sse,
    nb_kernel201_x86_64_sse,
    nb_kernel202_x86_64_sse,
    nb_kernel203_x86_64_sse,
    nb_kernel204_x86_64_sse,
    nb_kernel210_x86_64_sse,
    nb_kernel211_x86_64_sse,
    nb_kernel212_x86_64_sse,
    nb_kernel213_x86_64_sse,
    nb_kernel214_x86_64_sse,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    nb_kernel230_x86_64_sse,
    nb_kernel231_x86_64_sse,
    nb_kernel232_x86_64_sse,
    nb_kernel233_x86_64_sse,
    nb_kernel234_x86_64_sse,
    nb_kernel300_x86_64_sse,
    nb_kernel301_x86_64_sse,
    nb_kernel302_x86_64_sse,
    nb_kernel303_x86_64_sse,
    nb_kernel304_x86_64_sse,
    nb_kernel310_x86_64_sse,
    nb_kernel311_x86_64_sse,
    nb_kernel312_x86_64_sse,
    nb_kernel313_x86_64_sse,
    nb_kernel314_x86_64_sse,
    NULL,
    NULL,
    NULL,
    NULL,
    NULL,
    nb_kernel330_x86_64_sse,
    nb_kernel331_x86_64_sse,
    nb_kernel332_x86_64_sse,
    nb_kernel333_x86_64_sse,
    nb_kernel334_x86_64_sse,
    nb_kernel400_x86_64_sse,
    nb_kernel410_x86_64_sse,
    nb_kernel430_x86_64_sse
};


#ifdef GMX_THREADS
static gmx_thread_mutex_t 
nb_kernel_x86_64_sse_test_mutex = GMX_THREAD_MUTEX_INITIALIZER;
#endif


/*! Posix long jump label */
static jmp_buf         
nb_kernel_x86_64_sse_testprog;

/*! Result of x86_64 SSE test */
static bool      
nb_kernel_x86_64_sse_present;


static void 
nb_kernel_x86_64_sse_sigill_handler(int n)
{
  nb_kernel_x86_64_sse_present=FALSE;
  longjmp(nb_kernel_x86_64_sse_testprog,n);
}



/* Return 0 if SSE support is present, or
 * non-zero on failure.
 */
int 
nb_kernel_x86_64_sse_test(FILE *                log)
{
	/* 
	 * This should NOT be called from threads, 
	 * but just in case you still try to do it...
	 */
#ifdef GMX_THREADS
	gmx_thread_mutex_lock(&nb_kernel_x86_64_sse_test_mutex);
#endif
    
    if(log)
        fprintf(log,"Testing x86_64 SSE support...");

	nb_kernel_x86_64_sse_present = TRUE;
	signal(SIGILL,nb_kernel_x86_64_sse_sigill_handler);

	/* return to this point after executing the signal handler
	 * if we catch a SIGILL
	 */
	setjmp(nb_kernel_x86_64_sse_testprog); 

	if(nb_kernel_x86_64_sse_present)
		nb_kernel_x86_64_sse_test_asm();
	
	/* If SSE worked, then success is still 1.
     * If we got SIGILL, it was set to 0 in sigill_handler().
     */

	if(log)
		fprintf(log," %spresent.\n", 
				nb_kernel_x86_64_sse_present ? "":"not ");
	
#ifdef GMX_THREADS
	gmx_thread_mutex_unlock(&nb_kernel_x86_64_sse_test_mutex);
#endif
    
	return ((nb_kernel_x86_64_sse_present) ? 0 : -1);
}

				
			

void
nb_kernel_setup_x86_64_sse(FILE *log,nb_kernel_t **list)
{
    int i;
    nb_kernel_t *p;
    
    if(nb_kernel_x86_64_sse_test(log) != 0)
        return;
    
    for(i=0;i<eNR_NBKERNEL_NR;i++)
    {
        p = kernellist_x86_64_sse[i];
        if(p!=NULL)
            list[i] = p; 
    }
}