src/parsec/disk-image/parsec/parsec-benchmark/pkgs/libs/tbblib/src/src/test/test_model_plugin.cpp - public/gem5-resources - Git at Google

 /*
     Copyright 2005-2010 Intel Corporation.  All Rights Reserved.

     This file is part of Threading Building Blocks.

     Threading Building Blocks is free software; you can redistribute it
     and/or modify it under the terms of the GNU General Public License
     version 2 as published by the Free Software Foundation.

     Threading Building Blocks is distributed in the hope that it will be
     useful, but WITHOUT ANY WARRANTY; without even the implied warranty
     of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.

     You should have received a copy of the GNU General Public License
     along with Threading Building Blocks; if not, write to the Free Software
     Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

     As a special exception, you may use this file as part of a free software
     library without restriction.  Specifically, if other files instantiate
     templates or use macros or inline functions from this file, or you compile
     this file and link it with other files to produce an executable, this
     file does not by itself cause the resulting executable to be covered by
     the GNU General Public License.  This exception does not however
     invalidate any other reasons why the executable file might be covered by
     the GNU General Public License.
 */


 #if _WIN32 || _WIN64
 #include <windows.h>
 #else
 #include <dlfcn.h>
 #endif

 #include <stdlib.h>
 #include <stdio.h>

 #include "tbb/tbb_config.h"

 #if !TBB_USE_EXCEPTIONS && _MSC_VER
     // Suppress "C++ exception handler used, but unwind semantics are not enabled" warning in STL headers
     #pragma warning (push)
     #pragma warning (disable: 4530)
 #endif

 #include <stdexcept>

 #if !TBB_USE_EXCEPTIONS && _MSC_VER
     #pragma warning (pop)
 #endif

 #if TBB_USE_EXCEPTIONS
     #include "harness_report.h"
 #endif

 #ifdef _USRDLL
 #include "tbb/task_scheduler_init.h"

 class CModel {
 public:
     CModel(void) {};
     static tbb::task_scheduler_init tbb_init;

     void init_and_terminate( int );
 };

 tbb::task_scheduler_init CModel::tbb_init(1);

 //! Test that task::initialize and task::terminate work when doing nothing else.
 /** maxthread is treated as the "maximum" number of worker threads. */
 void CModel::init_and_terminate( int maxthread ) {
     for( int i=0; i<200; ++i ) {
         switch( i&3 ) {
             default: {
                 tbb::task_scheduler_init init( rand() % maxthread + 1 );
                 break;
             }
             case 0: {
                 tbb::task_scheduler_init init;
                 break;
             }
             case 1: {
                 tbb::task_scheduler_init init( tbb::task_scheduler_init::automatic );
                 break;
             }
             case 2: {
                 tbb::task_scheduler_init init( tbb::task_scheduler_init::deferred );
                 init.initialize( rand() % maxthread + 1 );
                 init.terminate();
                 break;
             }
         }
     }
 }

 extern "C"
 #if _WIN32 || _WIN64
 __declspec(dllexport)
 #endif
 void plugin_call(int maxthread)
 {
     srand(2);
     __TBB_TRY {
         CModel model;
         model.init_and_terminate(maxthread);
     } __TBB_CATCH( std::runtime_error& error ) {
 #if TBB_USE_EXCEPTIONS
         REPORT("ERROR: %s\n", error.what());
 #endif /* TBB_USE_EXCEPTIONS */
     }
 }

 #else /* _USRDLL undefined */

 #define HARNESS_NO_ASSERT 1
 #include "harness.h"

 extern "C" void plugin_call(int);

 void report_error_in(const char* function_name)
 {
 #if _WIN32 || _WIN64
     char* message;
     int code = GetLastError();

     FormatMessage(
         FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
         NULL, code,MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
         (char*)&message, 0, NULL );
 #else
     char* message = (char*)dlerror();
     int code = 0;
 #endif
     REPORT( "%s failed with error %d: %s\n", function_name, code, message);

 #if _WIN32 || _WIN64
     LocalFree(message);
 #endif
 }

 int use_lot_of_tls() {
     int count = 0;
 #if _WIN32 || _WIN64
     DWORD last_handles[10];
     DWORD result;
     result = TlsAlloc();
     while( result!=TLS_OUT_OF_INDEXES ) {
         last_handles[++count%10] = result;
         result = TlsAlloc();
     }
     for( int i=0; i<10; ++i )
         TlsFree(last_handles[i]);
 #else
     pthread_key_t last_handles[10];
     pthread_key_t result;
     int setspecific_dummy=10;
     while( pthread_key_create(&result, NULL)==0
            && count < 4096 ) // Sun Solaris doesn't have any built-in limit, so we set something big enough
     {
         last_handles[++count%10] = result;
         pthread_setspecific(result,&setspecific_dummy);
     }
     REMARK("Created %d keys\n", count);
     for( int i=0; i<10; ++i )
         pthread_key_delete(last_handles[i]);
 #endif
     return count-10;
 }

 typedef void (*PLUGIN_CALL)(int);

 int TestMain () {
 #if !RML_USE_WCRM
     PLUGIN_CALL my_plugin_call;

     int tls_key_count = use_lot_of_tls();
     REMARK("%d thread local objects allocated in advance\n", tls_key_count);

 #if _WIN32 || _WIN64
     HMODULE hLib;
 #if __TBB_ARENA_PER_MASTER
     hLib = LoadLibrary("irml.dll");
     if ( !hLib )
         hLib = LoadLibrary("irml_debug.dll");
     if ( !hLib )
         return Harness::Skipped; // No shared RML, skip the test
     FreeLibrary(hLib);
 #endif /* __TBB_ARENA_PER_MASTER */
 #else /* !WIN */
 #if __APPLE__
     #define LIBRARY_NAME(base) base".dylib"
 #else
     #define LIBRARY_NAME(base) base".so"
 #endif
     void* hLib;
 #if __TBB_ARENA_PER_MASTER
 #if __linux__
     #define RML_LIBRARY_NAME(base) LIBRARY_NAME(base) ".1"
 #else
     #define RML_LIBRARY_NAME(base) LIBRARY_NAME(base)
 #endif
     hLib = dlopen(RML_LIBRARY_NAME("libirml"), RTLD_LAZY);
     if ( !hLib )
         hLib = dlopen(RML_LIBRARY_NAME("libirml_debug"), RTLD_LAZY);
     if ( !hLib )
         return Harness::Skipped;
     dlclose(hLib);
 #endif /* __TBB_ARENA_PER_MASTER */
 #endif /* OS */
     for( int i=1; i<100; ++i ) {
         REMARK("Iteration %d, loading plugin library...\n", i);
 #if _WIN32 || _WIN64
         hLib = LoadLibrary("test_model_plugin_dll.dll");
         if ( !hLib ) {
 #if !__TBB_NO_IMPLICIT_LINKAGE
             report_error_in("LoadLibrary");
             return -1;
 #else
             return Harness::Skipped;
 #endif
         }
         my_plugin_call = (PLUGIN_CALL) GetProcAddress(hLib, "plugin_call");
         if (my_plugin_call==NULL) {
             report_error_in("GetProcAddress");
             return -1;
         }
 #else /* !WIN */
         hLib = dlopen( LIBRARY_NAME("test_model_plugin_dll"), RTLD_LAZY );
         if ( !hLib ) {
 #if !__TBB_NO_IMPLICIT_LINKAGE
             report_error_in("dlopen");
             return -1;
 #else
             return Harness::Skipped;
 #endif
         }
         my_plugin_call = PLUGIN_CALL (dlsym(hLib, "plugin_call"));
         if (my_plugin_call==NULL) {
             report_error_in("dlsym");
             return -1;
         }
 #endif /* !WIN */

         REMARK("Calling plugin method...\n");
         my_plugin_call(MaxThread);

         REMARK("Unloading plugin library...\n");
 #if _WIN32 || _WIN64
         FreeLibrary(hLib);
 #else
         dlclose(hLib);
 #endif
     } // end for(1,100)

     return Harness::Done;
 #else
     return Harness::Skipped;
 #endif /* !RML_USE_WCRM */
 }

 #endif//_USRDLL
	/*
	Copyright 2005-2010 Intel Corporation. All Rights Reserved.

	This file is part of Threading Building Blocks.

	Threading Building Blocks is free software; you can redistribute it
	and/or modify it under the terms of the GNU General Public License
	version 2 as published by the Free Software Foundation.

	Threading Building Blocks is distributed in the hope that it will be
	useful, but WITHOUT ANY WARRANTY; without even the implied warranty
	of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with Threading Building Blocks; if not, write to the Free Software
	Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA

	As a special exception, you may use this file as part of a free software
	library without restriction. Specifically, if other files instantiate
	templates or use macros or inline functions from this file, or you compile
	this file and link it with other files to produce an executable, this
	file does not by itself cause the resulting executable to be covered by
	the GNU General Public License. This exception does not however
	invalidate any other reasons why the executable file might be covered by
	the GNU General Public License.
	*/


	#if _WIN32 \|\| _WIN64
	#include <windows.h>
	#else
	#include <dlfcn.h>
	#endif

	#include <stdlib.h>
	#include <stdio.h>

	#include "tbb/tbb_config.h"

	#if !TBB_USE_EXCEPTIONS && _MSC_VER
	// Suppress "C++ exception handler used, but unwind semantics are not enabled" warning in STL headers
	#pragma warning (push)
	#pragma warning (disable: 4530)
	#endif

	#include <stdexcept>

	#if !TBB_USE_EXCEPTIONS && _MSC_VER
	#pragma warning (pop)
	#endif

	#if TBB_USE_EXCEPTIONS
	#include "harness_report.h"
	#endif

	#ifdef _USRDLL
	#include "tbb/task_scheduler_init.h"

	class CModel {
	public:
	CModel(void) {};
	static tbb::task_scheduler_init tbb_init;

	void init_and_terminate( int );
	};

	tbb::task_scheduler_init CModel::tbb_init(1);

	//! Test that task::initialize and task::terminate work when doing nothing else.
	/** maxthread is treated as the "maximum" number of worker threads. */
	void CModel::init_and_terminate( int maxthread ) {
	for( int i=0; i<200; ++i ) {
	switch( i&3 ) {
	default: {
	tbb::task_scheduler_init init( rand() % maxthread + 1 );
	break;
	}
	case 0: {
	tbb::task_scheduler_init init;
	break;
	}
	case 1: {
	tbb::task_scheduler_init init( tbb::task_scheduler_init::automatic );
	break;
	}
	case 2: {
	tbb::task_scheduler_init init( tbb::task_scheduler_init::deferred );
	init.initialize( rand() % maxthread + 1 );
	init.terminate();
	break;
	}
	}
	}
	}

	extern "C"
	#if _WIN32 \|\| _WIN64
	__declspec(dllexport)
	#endif
	void plugin_call(int maxthread)
	{
	srand(2);
	__TBB_TRY {
	CModel model;
	model.init_and_terminate(maxthread);
	} __TBB_CATCH( std::runtime_error& error ) {
	#if TBB_USE_EXCEPTIONS
	REPORT("ERROR: %s\n", error.what());
	#endif /* TBB_USE_EXCEPTIONS */
	}
	}

	#else /* _USRDLL undefined */

	#define HARNESS_NO_ASSERT 1
	#include "harness.h"

	extern "C" void plugin_call(int);

	void report_error_in(const char* function_name)
	{
	#if _WIN32 \|\| _WIN64
	char* message;
	int code = GetLastError();

	FormatMessage(
	FORMAT_MESSAGE_ALLOCATE_BUFFER \| FORMAT_MESSAGE_FROM_SYSTEM,
	NULL, code,MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
	(char*)&message, 0, NULL );
	#else
	char* message = (char*)dlerror();
	int code = 0;
	#endif
	REPORT( "%s failed with error %d: %s\n", function_name, code, message);

	#if _WIN32 \|\| _WIN64
	LocalFree(message);
	#endif
	}

	int use_lot_of_tls() {
	int count = 0;
	#if _WIN32 \|\| _WIN64
	DWORD last_handles[10];
	DWORD result;
	result = TlsAlloc();
	while( result!=TLS_OUT_OF_INDEXES ) {
	last_handles[++count%10] = result;
	result = TlsAlloc();
	}
	for( int i=0; i<10; ++i )
	TlsFree(last_handles[i]);
	#else
	pthread_key_t last_handles[10];
	pthread_key_t result;
	int setspecific_dummy=10;
	while( pthread_key_create(&result, NULL)==0
	&& count < 4096 ) // Sun Solaris doesn't have any built-in limit, so we set something big enough
	{
	last_handles[++count%10] = result;
	pthread_setspecific(result,&setspecific_dummy);
	}
	REMARK("Created %d keys\n", count);
	for( int i=0; i<10; ++i )
	pthread_key_delete(last_handles[i]);
	#endif
	return count-10;
	}

	typedef void (*PLUGIN_CALL)(int);

	int TestMain () {
	#if !RML_USE_WCRM
	PLUGIN_CALL my_plugin_call;

	int tls_key_count = use_lot_of_tls();
	REMARK("%d thread local objects allocated in advance\n", tls_key_count);

	#if _WIN32 \|\| _WIN64
	HMODULE hLib;
	#if __TBB_ARENA_PER_MASTER
	hLib = LoadLibrary("irml.dll");
	if ( !hLib )
	hLib = LoadLibrary("irml_debug.dll");
	if ( !hLib )
	return Harness::Skipped; // No shared RML, skip the test
	FreeLibrary(hLib);
	#endif /* __TBB_ARENA_PER_MASTER */
	#else /* !WIN */
	#if __APPLE__
	#define LIBRARY_NAME(base) base".dylib"
	#else
	#define LIBRARY_NAME(base) base".so"
	#endif
	void* hLib;
	#if __TBB_ARENA_PER_MASTER
	#if __linux__
	#define RML_LIBRARY_NAME(base) LIBRARY_NAME(base) ".1"
	#else
	#define RML_LIBRARY_NAME(base) LIBRARY_NAME(base)
	#endif
	hLib = dlopen(RML_LIBRARY_NAME("libirml"), RTLD_LAZY);
	if ( !hLib )
	hLib = dlopen(RML_LIBRARY_NAME("libirml_debug"), RTLD_LAZY);
	if ( !hLib )
	return Harness::Skipped;
	dlclose(hLib);
	#endif /* __TBB_ARENA_PER_MASTER */
	#endif /* OS */
	for( int i=1; i<100; ++i ) {
	REMARK("Iteration %d, loading plugin library...\n", i);
	#if _WIN32 \|\| _WIN64
	hLib = LoadLibrary("test_model_plugin_dll.dll");
	if ( !hLib ) {
	#if !__TBB_NO_IMPLICIT_LINKAGE
	report_error_in("LoadLibrary");
	return -1;
	#else
	return Harness::Skipped;
	#endif
	}
	my_plugin_call = (PLUGIN_CALL) GetProcAddress(hLib, "plugin_call");
	if (my_plugin_call==NULL) {
	report_error_in("GetProcAddress");
	return -1;
	}
	#else /* !WIN */
	hLib = dlopen( LIBRARY_NAME("test_model_plugin_dll"), RTLD_LAZY );
	if ( !hLib ) {
	#if !__TBB_NO_IMPLICIT_LINKAGE
	report_error_in("dlopen");
	return -1;
	#else
	return Harness::Skipped;
	#endif
	}
	my_plugin_call = PLUGIN_CALL (dlsym(hLib, "plugin_call"));
	if (my_plugin_call==NULL) {
	report_error_in("dlsym");
	return -1;
	}
	#endif /* !WIN */

	REMARK("Calling plugin method...\n");
	my_plugin_call(MaxThread);

	REMARK("Unloading plugin library...\n");
	#if _WIN32 \|\| _WIN64
	FreeLibrary(hLib);
	#else
	dlclose(hLib);
	#endif
	} // end for(1,100)

	return Harness::Done;
	#else
	return Harness::Skipped;
	#endif /* !RML_USE_WCRM */
	}

	#endif//_USRDLL