kfdtest: Add kfdtest source code
The code is a snapshot up to this commit around July 31 2018.
commit b00fadff36a3
Author: xinhui pan <xinhui.pan@amd.com>
Date: Mon Jul 30 09:53:03 2018 +0800
kfdtest: skip MMapLarge test on apu
Change-Id: I40e9a5a18e5c8f075e5290bb80532f1a3f689058
Signed-off-by: Yong Zhao <yong.zhao@amd.com>
[ROCm/ROCR-Runtime commit: 6df62c78b8]
Αυτή η υποβολή περιλαμβάνεται σε:
Yong Zhao
@@ -0,0 +1,240 @@
|
||||
/*
|
||||
* Copyright (C) 2014-2018 Advanced Micro Devices, Inc. All Rights Reserved.
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a
|
||||
* copy of this software and associated documentation files (the "Software"),
|
||||
* to deal in the Software without restriction, including without limitation
|
||||
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||||
* and/or sell copies of the Software, and to permit persons to whom the
|
||||
* Software is furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in
|
||||
* all copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||||
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
|
||||
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
|
||||
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
||||
* OTHER DEALINGS IN THE SOFTWARE.
|
||||
*
|
||||
*/
|
||||
|
||||
#ifndef WIN32
|
||||
|
||||
#include "OSWrapper.hpp"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include <unistd.h>
|
||||
#include <errno.h>
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <getopt.h>
|
||||
#include <drm.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <fcntl.h>
|
||||
#include <string.h>
|
||||
#include <sys/mman.h>
|
||||
#include <sys/ioctl.h>
|
||||
#include <sys/time.h>
|
||||
|
||||
static int protection_flags[8] = {int(PROT_NONE), int(PROT_READ), int(PROT_WRITE), int(PROT_READ | PROT_WRITE), int(PROT_EXEC), int(PROT_EXEC | PROT_READ), int(PROT_EXEC | PROT_WRITE), int(PROT_EXEC | PROT_WRITE | PROT_READ)};
|
||||
|
||||
void SetConsoleTextColor(TEXTCOLOR color) {
|
||||
// TODO complete
|
||||
}
|
||||
|
||||
void Delay(int delayCount) {
|
||||
// usleeps accept time in microseconds
|
||||
usleep(delayCount * 1000);
|
||||
}
|
||||
|
||||
void *VirtualAllocMemory(void *address, unsigned int size, int memProtection ) {
|
||||
void *ptr;
|
||||
|
||||
ptr = mmap(address, size, protection_flags[memProtection], MAP_PRIVATE|MAP_ANONYMOUS, -1, 0);
|
||||
|
||||
if (ptr == MAP_FAILED)
|
||||
ptr = NULL;
|
||||
return ptr;
|
||||
}
|
||||
|
||||
bool VirtualFreeMemory(void *address, unsigned int size) {
|
||||
if (munmap(address, size) == 0)
|
||||
return true;
|
||||
else
|
||||
return false;
|
||||
}
|
||||
|
||||
unsigned long GetLastErrorNo() {
|
||||
return errno;
|
||||
}
|
||||
|
||||
bool MultiProcessTest(const char *testToRun, int numOfProcesses, int runsPerProcess) {
|
||||
// TODO IMPLEMENT
|
||||
return false;
|
||||
}
|
||||
|
||||
unsigned long long GetSystemTickCountInMicroSec() {
|
||||
struct timeval t;
|
||||
gettimeofday(&t, 0);
|
||||
return t.tv_sec * 1000000ULL + t.tv_usec;
|
||||
}
|
||||
|
||||
bool SuspendAndWakeUp() {
|
||||
// TODO IMPLEMENT
|
||||
return false;
|
||||
}
|
||||
|
||||
void AcquirePrivilege(OS_PRIVILEGE priv) {
|
||||
// TODO IMPLEMENT
|
||||
}
|
||||
|
||||
void DisableKfd() {
|
||||
// TODO IMPLEMENT
|
||||
}
|
||||
|
||||
void EnableKfd() {
|
||||
// TODO IMPLEMENT
|
||||
}
|
||||
|
||||
bool ReadDriverConfigValue(CONFIG_VALUE config, unsigned int& rValue) {
|
||||
return false;
|
||||
}
|
||||
|
||||
void ComandLineArgumentsUsage() {
|
||||
printf("Invalid option value\n");
|
||||
printf("\t--hws arg\t - Force HW capability\n");
|
||||
printf("\t--profile arg\t - Test profile\n");
|
||||
printf("\t--child arg\t - Child Process\n");
|
||||
printf("\t--timeout arg\t - Time Out\n");
|
||||
printf("\t--dst_node\t - For testing multiple nodes");
|
||||
}
|
||||
|
||||
bool GetCommandLineArguments(int argc, char **argv, CommandLineArguments& rArgs) {
|
||||
int option_index = 0;
|
||||
|
||||
/* Make getop silent */
|
||||
opterr = 0;
|
||||
static struct option long_options[] = {
|
||||
{ "hws", required_argument, 0, 0 },
|
||||
{ "profile", required_argument, 0, 0},
|
||||
{ "child", required_argument, 0, 0},
|
||||
{ "timeout", required_argument, 0, 0},
|
||||
{ "node", required_argument, 0, 0 },
|
||||
{ "dst_node", required_argument, 0, 0 },
|
||||
{ 0, 0, 0, 0 }
|
||||
};
|
||||
|
||||
rArgs.HwsEnabled = HWCAP__DEFAULT;
|
||||
rArgs.TestProfile = TESTPROFILE_RUNALL;
|
||||
rArgs.ChildProcess = false;
|
||||
rArgs.TimeOut = 0;
|
||||
rArgs.NodeId = -1;
|
||||
rArgs.DstNodeId = -1;
|
||||
|
||||
while (true) {
|
||||
int c = getopt_long(argc, argv, "", long_options, &option_index);
|
||||
|
||||
/* Detect the end of the options. */
|
||||
if (c != 0)
|
||||
break;
|
||||
|
||||
/* If this option set a flag, do nothing else now. */
|
||||
if (long_options[option_index].flag != 0)
|
||||
continue;
|
||||
|
||||
if (optarg == NULL) {
|
||||
ComandLineArgumentsUsage();
|
||||
return false;
|
||||
}
|
||||
|
||||
switch (option_index) {
|
||||
/* HWS case */
|
||||
case 0:
|
||||
if (!strcmp(optarg, "disable")) {
|
||||
rArgs.HwsEnabled = HWCAP__FORCE_DISABLED;
|
||||
} else if (!strcmp(optarg, "enable")) {
|
||||
rArgs.HwsEnabled = HWCAP__FORCE_ENABLED;
|
||||
} else {
|
||||
ComandLineArgumentsUsage();
|
||||
return false;
|
||||
}
|
||||
break;
|
||||
/* TEST PROFILE */
|
||||
case 1:
|
||||
if (!strcmp(optarg, "dev")) {
|
||||
rArgs.TestProfile = TESTPROFILE_DEV;
|
||||
} else if (!strcmp(optarg, "promo")) {
|
||||
rArgs.TestProfile = TESTPROFILE_PROMO;
|
||||
} else if (!strcmp(optarg, "all")) {
|
||||
rArgs.TestProfile = TESTPROFILE_RUNALL;
|
||||
} else {
|
||||
ComandLineArgumentsUsage();
|
||||
return false;
|
||||
}
|
||||
break;
|
||||
|
||||
case 2:
|
||||
rArgs.ChildProcess = true;
|
||||
break;
|
||||
|
||||
case 3:
|
||||
{
|
||||
int timeOut = atoi(optarg);
|
||||
if (timeOut > 0)
|
||||
rArgs.TimeOut = timeOut;
|
||||
}
|
||||
break;
|
||||
case 4:
|
||||
{
|
||||
int nodeId = atoi(optarg);
|
||||
if (nodeId >= 0)
|
||||
rArgs.NodeId = nodeId;
|
||||
}
|
||||
break;
|
||||
case 5:
|
||||
{
|
||||
int dstNodeId = atoi(optarg);
|
||||
if (dstNodeId >= 0)
|
||||
rArgs.DstNodeId = dstNodeId;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void HWMemoryBarrier() {
|
||||
__sync_synchronize();
|
||||
}
|
||||
|
||||
bool StartThread(unsigned int (*thread_func)(void*), void* param, uint64_t& thread_id) {
|
||||
pthread_t id;
|
||||
bool ret = false;
|
||||
typedef void* (*pthread_func_t)(void*);
|
||||
|
||||
if (!pthread_create(&id, NULL, (pthread_func_t)thread_func, param)) {
|
||||
thread_id = (pthread_t)id;
|
||||
ret = true;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
bool WaitForThread(uint64_t threadId) {
|
||||
return 0 == pthread_join((pthread_t)threadId, NULL);
|
||||
}
|
||||
|
||||
long AtomicInc(volatile long* pValue) {
|
||||
return __sync_add_and_fetch(pValue, 1);
|
||||
}
|
||||
|
||||
void MemoryBarrier() {
|
||||
__sync_synchronize();
|
||||
}
|
||||
|
||||
#endif // !WIN32
|
||||
Αναφορά σε νέο ζήτημα
Block a user