Skip to content

GPU testsuite with self-hosted runners #222

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
j143 wants to merge 4 commits into
base: main-gpu
Choose a base branch
from
Open

GPU testsuite with self-hosted runners #222

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
4c79519
[SYSTEMDS-3020] Initial GPU junit tests
corepointer Jun 11, 2021
1fe5b31
[MINOR] Various unit test related small corrections and improvements
corepointer Jun 14, 2021
b4aea22
add basic action to test gpu hosted on runner
j143 Nov 5, 2021
1ae56c7
add flow trigger for push and pr event
j143 Nov 5, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 22 additions & 0 deletions .github/workflows/gpuTests.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
name: gpu tests

on:
push:
branches:
- main-gpu
pull_request:
branches:
- main-gpu
workflow_dispatch:

jobs:
test:
name: gpu ubuntu instance
runs-on: [self-hosted, linux, x64, gpu]

steps:
- name: Checkout Repository
uses: actions/checkout@v2

- name: print hello
run: printf "Hello GPU!"
10 changes: 7 additions & 3 deletions pom.xml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,9 @@
<java.level>1.8</java.level>
<!-->Testing settings<!-->
<maven.test.skip>true</maven.test.skip>
<test-parallel>classes</test-parallel>
<test-threadCount>2</test-threadCount>
<test-forkCount>1C</test-forkCount>
<rerun.failing.tests.count>2</rerun.failing.tests.count>
<jacoco.skip>true</jacoco.skip>
<jacoco.include>**</jacoco.include>
Expand Down Expand Up @@ -294,9 +297,10 @@
<version>3.0.0-M5</version>
<configuration>
<skipTests>${maven.test.skip}</skipTests>
<parallel>classes</parallel>
<threadCount>2</threadCount>
<forkCount>1C</forkCount>
<parallel>${test-parallel}</parallel>
<threadCount>${test-threadCount}</threadCount>
<!-- 1C means the number of threads times 1 possible maximum forks for testing-->
<forkCount>${test-forkCount}</forkCount>
<reuseForks>false</reuseForks>
<reportFormat>brief</reportFormat>
<trimStackTrace>true</trimStackTrace>
Expand Down
4 changes: 3 additions & 1 deletion src/test/java/org/apache/sysds/test/AutomatedTestBase.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1522,8 +1522,10 @@ else if(rtplatform == ExecMode.SPARK)

if(TEST_GPU)
args.add("-gpu");
if(VERBOSE_STATS)
if(VERBOSE_STATS) {
args.add("-stats");
args.add("100");
}
}

public static int getRandomAvailablePort() {
Expand Down
2 changes: 1 addition & 1 deletion src/test/java/org/apache/sysds/test/functions/dnn/Conv2DTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -269,7 +269,7 @@ public void runConv2DTest( ExecType et, int imgSize, int numImg, int numChannels

String RI_HOME = SCRIPT_DIR + TEST_DIR;
fullDMLScriptName = RI_HOME + TEST_NAME + ".dml";
programArgs = new String[] {"recompile_runtime", "-args",
programArgs = new String[] {"-explain", "recompile_runtime", "-args",
String.valueOf(imgSize), String.valueOf(numImg),
String.valueOf(numChannels), String.valueOf(numFilters),
String.valueOf(filterSize), String.valueOf(stride), String.valueOf(pad),
Expand Down
Loading