Skip to content

1. add examples of system test #10

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

Merged
TONYSKYZENG merged 2 commits into from
May 17, 2023
Merged

1. add examples of system test #10

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
52bdd49
1. add examples of system test
TONYSKYZENG May 17, 2023
b07064d
1. add examples of system test
TONYSKYZENG May 17, 2023
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
3 changes: 2 additions & 1 deletion .github/workflows/cmake.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,4 +44,5 @@ jobs:
# Execute tests defined by the CMake configuration.
# See https://cmake.org/cmake/help/latest/manual/ctest.1.html for more detail
run: |
./cpp_test "--success"
./cpp_test "--success"
./sketch_test "--success"
2 changes: 1 addition & 1 deletion CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ add_subdirectory(src)
# Add Library
get_sources(IntelliStream_SOURCE_FILES)
get_headers(IntelliStream_HEADER_FILES)
add_library(IntelliStream SHARED ${IntelliStream_SOURCE_FILES} ${IntelliStream_HEADER_FILES} ${CMAKE_CURRENT_BINARY_DIR} ${CMAKE_CURRENT_BINARY_DIR} )
add_library(IntelliStream SHARED ${IntelliStream_SOURCE_FILES} ${IntelliStream_HEADER_FILES} ${CMAKE_CURRENT_BINARY_DIR} ${CMAKE_CURRENT_BINARY_DIR})
set_property(TARGET IntelliStream PROPERTY CXX_STANDARD 20)

target_include_directories(IntelliStream PUBLIC "include")
Expand Down
2 changes: 1 addition & 1 deletion Doxyfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ PROJECT_NAME = "AMMBench"
# could be handy for archiving the generated documentation or if some version
# control system is used.

PROJECT_NUMBER = 0.1.0
PROJECT_NUMBER = 0.1.1

# Using the PROJECT_BRIEF tag one can provide an optional one line description
# for a project that appears at the top of each page and should give viewer a
Expand Down
4 changes: 2 additions & 2 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -198,6 +198,6 @@ You will find the figures then.
1. If you use Torch with cuda, the nvcc will refuse to work as it doesn't support c++20 yet. Therefore, we disabled the
global requirement check of C++ 20, and only leave an "-std=c++20" option for g++. This will be fixed after nvidia
can support c++20 in cuda.
2. Some pytorch version can not work well with liblog4cxx and googletest, so we diabled it.
2. Some pytorch version can not work well with liblog4cxx and googletest, so we diabled it.
3. Clion may fail to render and highlight the torch apis. In this case, kindly type a random line of "555"
to validate the highlight when you need it, and comment it during a compile. :)
to validate the highlight when you need it, and comment it during a compile. :)
27 changes: 16 additions & 11 deletions benchmark/scripts/scanARank/drawTogether.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -81,9 +81,12 @@ def readResultSingle(singleValue, resultPath):
cacheRefs = readConfig(resultFname, "cacheRefs")
return elapsedTime, cacheMiss, cacheRefs


def cleanPath(path):
os.system("sudo rm -rf " + path)
os.system("sudo mkdir " + path)


def readResultVector(singleValueVec, resultPath):
elapseTimeVec = []
cacheMissVec = []
Expand All @@ -99,14 +102,14 @@ def readResultVector(singleValueVec, resultPath):
def main():
exeSpace = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/"
resultPath = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag
resultPathFDAMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/FDAMM"
resultPathCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/CoFD"
resultPathBetaCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/BCoFD"
resultPathRAWMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/RAWMM"
resultPathFDAMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/FDAMM"
resultPathCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/CoFD"
resultPathBetaCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/BCoFD"
resultPathRAWMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/RAWMM"
figPath = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/figures/" + scanTag
configTemplate = exeSpace + "config.csv"
valueVec = [2,5,10,20,50,100,200,500,900]
valueVecRun = 1000-np.array(valueVec)
valueVec = [2, 5, 10, 20, 50, 100, 200, 500, 900]
valueVecRun = 1000 - np.array(valueVec)
print(configTemplate)
# run
if (len(sys.argv) < 2):
Expand All @@ -125,18 +128,20 @@ def main():
runScanVector(exeSpace, valueVecRun, resultPathCoFD, "config_CoAMM.csv")
runScanVector(exeSpace, valueVecRun, resultPathBetaCoFD, "config_BCoAMM.csv")
runScanVector(exeSpace, valueVecRun, resultPathRAWMM, "config_RAWMM.csv")
evaTypes=['FDAMM','MM','Co-FD','BCO-FD']
evaTypes = ['FDAMM', 'MM', 'Co-FD', 'BCO-FD']
elapseTimeVecFD, cacheMissVecFD, cacheRefVecFD = readResultVector(valueVecRun, resultPathFDAMM)
elapseTimeVecCoFD, cacheMissVecCoFD, cacheRefVecCoFD = readResultVector(valueVecRun, resultPathCoFD)
elapseTimeVeCB, cacheMissVecB, cacheRefVecB = readResultVector(valueVecRun, resultPathBetaCoFD)
elapseTimeVecRAW, cacheMissVecRAW, cacheRefVecRAW = readResultVector(valueVecRun, resultPathRAWMM)
# os.system("mkdir " + figPath)
groupLine.DrawFigureYnormal([valueVec,valueVec,valueVec,valueVec], [elapseTimeVecFD,elapseTimeVecRAW,elapseTimeVecCoFD,elapseTimeVeCB],
groupLine.DrawFigureYnormal([valueVec, valueVec, valueVec, valueVec],
[elapseTimeVecFD, elapseTimeVecRAW, elapseTimeVecCoFD, elapseTimeVeCB],
evaTypes,
"Rank of matrix A", "elapsed time (ms)", 0, 1, figPath + "Rank" + "_elapsedTime",
True)
groupLine.DrawFigureYnormal([valueVec,valueVec,valueVec,valueVec], [cacheMissVecFD/cacheRefVecFD*100.0,cacheMissVecRAW/cacheRefVecRAW*100.0,
cacheMissVecCoFD/cacheRefVecCoFD*100.0,cacheMissVecB/cacheRefVecB*100.0],
groupLine.DrawFigureYnormal([valueVec, valueVec, valueVec, valueVec],
[cacheMissVecFD / cacheRefVecFD * 100.0, cacheMissVecRAW / cacheRefVecRAW * 100.0,
cacheMissVecCoFD / cacheRefVecCoFD * 100.0, cacheMissVecB / cacheRefVecB * 100.0],
evaTypes,
"Rank of matrix A", "cacheMiss (%)", 0, 1, figPath + "Rank" + "_cacheMiss",
True)
Expand All @@ -146,7 +151,7 @@ def main():
# groupLine.DrawFigureYnormal([singleValueVec,singleValueVec],[errVec,aqpErrVec],['w/o aqp',"w/ MeanAqp"],"watermark time (ms)","Error",0,1,figPath+"wm_MeanAqp",True)
# print(errVec)
# print(aqpErrVec)
#print(elapseTimeVecFD)
# print(elapseTimeVecFD)
# readResultsingleValue(50,resultPath)


Expand Down
6 changes: 4 additions & 2 deletions benchmark/scripts/scanARank/groupLine.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ def DrawFigure(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max,

# you may control the limits on your own.

#plt.ylim(y_min, y_max)
# plt.ylim(y_min, y_max)

plt.grid(axis='y', color='gray')
figure.yaxis.set_major_locator(LogLocator(base=10))
Expand All @@ -150,7 +150,7 @@ def DrawFigure(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max,
size = fig.get_size_inches()
dpi = fig.get_dpi()

#plt.show()
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


Expand Down Expand Up @@ -208,6 +208,7 @@ def DrawFigureXYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min,
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


# draw a line chart
def DrawFigureYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max, filename, allow_legend):
# you may change the figure size on your own.
Expand Down Expand Up @@ -262,6 +263,7 @@ def DrawFigureYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min,
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


# example for reading csv file
def ReadFile():
y = []
Expand Down
31 changes: 18 additions & 13 deletions benchmark/scripts/scanARow/drawTogether.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -81,9 +81,12 @@ def readResultSingle(singleValue, resultPath):
cacheRefs = readConfig(resultFname, "cacheRefs")
return elapsedTime, cacheMiss, cacheRefs


def cleanPath(path):
os.system("sudo rm -rf " + path)
os.system("sudo mkdir " + path)


def readResultVector(singleValueVec, resultPath):
elapseTimeVec = []
cacheMissVec = []
Expand All @@ -99,13 +102,13 @@ def readResultVector(singleValueVec, resultPath):
def main():
exeSpace = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/"
resultPath = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag
resultPathFDAMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/FDAMM"
resultPathCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/CoFD"
resultPathBetaCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/BCoFD"
resultPathRAWMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/RAWMM"
resultPathFDAMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/FDAMM"
resultPathCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/CoFD"
resultPathBetaCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/BCoFD"
resultPathRAWMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/RAWMM"
figPath = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/figures/" + scanTag
configTemplate = exeSpace + "config.csv"
valueVec = [100, 200, 500, 1000, 2000, 5000, 10000, 20000,50000]
valueVec = [100, 200, 500, 1000, 2000, 5000, 10000, 20000, 50000]
valueVecDisp = np.array(valueVec)
print(configTemplate)
# run
Expand All @@ -125,18 +128,20 @@ def main():
runScanVector(exeSpace, valueVec, resultPathCoFD, "config_CoAMM.csv")
runScanVector(exeSpace, valueVec, resultPathBetaCoFD, "config_BCoAMM.csv")
runScanVector(exeSpace, valueVec, resultPathRAWMM, "config_RAWMM.csv")
evaTypes=['FDAMM','MM','Co-FD','BCO-FD']
evaTypes = ['FDAMM', 'MM', 'Co-FD', 'BCO-FD']
elapseTimeVecFD, cacheMissVecFD, cacheRefVecFD = readResultVector(valueVec, resultPathFDAMM)
elapseTimeVecCoFD, cacheMissVecCoFD, cacheRefVecCoFD = readResultVector(valueVec, resultPathCoFD)
elapseTimeVeCB, cacheMissVecB, cacheRefVecB = readResultVector(valueVec, resultPathBetaCoFD)
elapseTimeVecRAW, cacheMissVecRAW, cacheRefVecRAW = readResultVector(valueVec, resultPathRAWMM)
# os.system("mkdir " + figPath)
groupLine.DrawFigure([valueVec,valueVec,valueVec,valueVec], [elapseTimeVecFD,elapseTimeVecRAW,elapseTimeVecCoFD,elapseTimeVeCB],
evaTypes,
"#elements in A's row", "elapsed time (ms)", 0, 1, figPath + scanTag + "_elapsedTime",
True)
groupLine.DrawFigureYnormal([valueVec,valueVec,valueVec,valueVec], [cacheMissVecFD/cacheRefVecFD*100.0,cacheMissVecRAW/cacheRefVecRAW*100.0,
cacheMissVecCoFD/cacheRefVecCoFD*100.0,cacheMissVecB/cacheRefVecB*100.0],
groupLine.DrawFigure([valueVec, valueVec, valueVec, valueVec],
[elapseTimeVecFD, elapseTimeVecRAW, elapseTimeVecCoFD, elapseTimeVeCB],
evaTypes,
"#elements in A's row", "elapsed time (ms)", 0, 1, figPath + scanTag + "_elapsedTime",
True)
groupLine.DrawFigureYnormal([valueVec, valueVec, valueVec, valueVec],
[cacheMissVecFD / cacheRefVecFD * 100.0, cacheMissVecRAW / cacheRefVecRAW * 100.0,
cacheMissVecCoFD / cacheRefVecCoFD * 100.0, cacheMissVecB / cacheRefVecB * 100.0],
evaTypes,
"#elements in A's row", "cacheMiss (%)", 0, 1, figPath + scanTag + "_cacheMiss",
True)
Expand All @@ -146,7 +151,7 @@ def main():
# groupLine.DrawFigureYnormal([singleValueVec,singleValueVec],[errVec,aqpErrVec],['w/o aqp',"w/ MeanAqp"],"watermark time (ms)","Error",0,1,figPath+"wm_MeanAqp",True)
# print(errVec)
# print(aqpErrVec)
#print(elapseTimeVecFD)
# print(elapseTimeVecFD)
# readResultsingleValue(50,resultPath)


Expand Down
6 changes: 4 additions & 2 deletions benchmark/scripts/scanARow/groupLine.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ def DrawFigure(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max,

# you may control the limits on your own.

#plt.ylim(y_min, y_max)
# plt.ylim(y_min, y_max)

plt.grid(axis='y', color='gray')
figure.yaxis.set_major_locator(LogLocator(base=10))
Expand All @@ -150,7 +150,7 @@ def DrawFigure(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max,
size = fig.get_size_inches()
dpi = fig.get_dpi()

#plt.show()
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


Expand Down Expand Up @@ -208,6 +208,7 @@ def DrawFigureXYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min,
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


# draw a line chart
def DrawFigureYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max, filename, allow_legend):
# you may change the figure size on your own.
Expand Down Expand Up @@ -262,6 +263,7 @@ def DrawFigureYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min,
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


# example for reading csv file
def ReadFile():
y = []
Expand Down
37 changes: 21 additions & 16 deletions benchmark/scripts/scanASparse/drawTogether.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -81,9 +81,12 @@ def readResultSingle(singleValue, resultPath):
cacheRefs = readConfig(resultFname, "cacheRefs")
return elapsedTime, cacheMiss, cacheRefs


def cleanPath(path):
os.system("sudo rm -rf " + path)
os.system("sudo mkdir " + path)


def readResultVector(singleValueVec, resultPath):
elapseTimeVec = []
cacheMissVec = []
Expand All @@ -99,13 +102,13 @@ def readResultVector(singleValueVec, resultPath):
def main():
exeSpace = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/"
resultPath = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag
resultPathFDAMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/FDAMM"
resultPathCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/CoFD"
resultPathBetaCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/BCoFD"
resultPathRAWMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag+"/RAWMM"
resultPathFDAMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/FDAMM"
resultPathCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/CoFD"
resultPathBetaCoFD = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/BCoFD"
resultPathRAWMM = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/results/" + scanTag + "/RAWMM"
figPath = os.path.abspath(os.path.join(os.getcwd(), "../..")) + "/figures/" + scanTag
configTemplate = exeSpace + "config.csv"
valueVec = [0.1,0.2,0.3,0.4,0.5,0.6,0.7,0.8,0.9,1.0]
valueVec = [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
valueVecDisp = np.array(valueVec)
print(configTemplate)
# run
Expand All @@ -125,28 +128,30 @@ def main():
runScanVector(exeSpace, valueVec, resultPathCoFD, "config_CoAMM.csv")
runScanVector(exeSpace, valueVec, resultPathBetaCoFD, "config_BCoAMM.csv")
runScanVector(exeSpace, valueVec, resultPathRAWMM, "config_RAWMM.csv")
evaTypes=['FDAMM','MM','Co-FD','BCO-FD']
evaTypes = ['FDAMM', 'MM', 'Co-FD', 'BCO-FD']
elapseTimeVecFD, cacheMissVecFD, cacheRefVecFD = readResultVector(valueVec, resultPathFDAMM)
elapseTimeVecCoFD, cacheMissVecCoFD, cacheRefVecCoFD = readResultVector(valueVec, resultPathCoFD)
elapseTimeVeCB, cacheMissVecB, cacheRefVecB = readResultVector(valueVec, resultPathBetaCoFD)
elapseTimeVecRAW, cacheMissVecRAW, cacheRefVecRAW = readResultVector(valueVec, resultPathRAWMM)
# os.system("mkdir " + figPath)
groupLine.DrawFigureXYnormal([valueVec,valueVec,valueVec,valueVec], [elapseTimeVecFD,elapseTimeVecRAW,elapseTimeVecCoFD,elapseTimeVeCB],
evaTypes,
"Density of matrix A", "elapsed time (ms)", 0, 1, figPath + scanTag + "_elapsedTime",
True)
groupLine.DrawFigureXYnormal([valueVec,valueVec,valueVec,valueVec], [cacheMissVecFD/cacheRefVecFD*100.0,cacheMissVecRAW/cacheRefVecRAW*100.0,
cacheMissVecCoFD/cacheRefVecCoFD*100.0,cacheMissVecB/cacheRefVecB*100.0],
evaTypes,
"Density of matrix A", "cacheMiss (%)", 0, 1, figPath + scanTag + "_cacheMiss",
True)
groupLine.DrawFigureXYnormal([valueVec, valueVec, valueVec, valueVec],
[elapseTimeVecFD, elapseTimeVecRAW, elapseTimeVecCoFD, elapseTimeVeCB],
evaTypes,
"Density of matrix A", "elapsed time (ms)", 0, 1, figPath + scanTag + "_elapsedTime",
True)
groupLine.DrawFigureXYnormal([valueVec, valueVec, valueVec, valueVec],
[cacheMissVecFD / cacheRefVecFD * 100.0, cacheMissVecRAW / cacheRefVecRAW * 100.0,
cacheMissVecCoFD / cacheRefVecCoFD * 100.0, cacheMissVecB / cacheRefVecB * 100.0],
evaTypes,
"Density of matrix A", "cacheMiss (%)", 0, 1, figPath + scanTag + "_cacheMiss",
True)
# draw2yLine("watermark time (ms)",singleValueVecDisp,lat95Vec,errVec,"95% Latency (ms)","Error","ms","",figPath+"wm_lat")
# draw2yLine("watermark time (ms)",singleValueVecDisp,thrVec,errVec,"Throughput (KTp/s)","Error","KTp/s","",figPath+"wm_thr")
# draw2yLine("watermark time (ms)",singleValueVecDisp,lat95Vec,compVec,"95% Latency (ms)","Completeness","ms","",figPath+"wm_omp")
# groupLine.DrawFigureYnormal([singleValueVec,singleValueVec],[errVec,aqpErrVec],['w/o aqp',"w/ MeanAqp"],"watermark time (ms)","Error",0,1,figPath+"wm_MeanAqp",True)
# print(errVec)
# print(aqpErrVec)
#print(elapseTimeVecFD)
# print(elapseTimeVecFD)
# readResultsingleValue(50,resultPath)


Expand Down
6 changes: 4 additions & 2 deletions benchmark/scripts/scanASparse/groupLine.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ def DrawFigure(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max,

# you may control the limits on your own.

#plt.ylim(y_min, y_max)
# plt.ylim(y_min, y_max)

plt.grid(axis='y', color='gray')
figure.yaxis.set_major_locator(LogLocator(base=10))
Expand All @@ -150,7 +150,7 @@ def DrawFigure(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max,
size = fig.get_size_inches()
dpi = fig.get_dpi()

#plt.show()
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


Expand Down Expand Up @@ -208,6 +208,7 @@ def DrawFigureXYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min,
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


# draw a line chart
def DrawFigureYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min, y_max, filename, allow_legend):
# you may change the figure size on your own.
Expand Down Expand Up @@ -262,6 +263,7 @@ def DrawFigureYnormal(xvalues, yvalues, legend_labels, x_label, y_label, y_min,
# plt.show()
plt.savefig(filename + ".pdf", bbox_inches='tight')


# example for reading csv file
def ReadFile():
y = []
Expand Down
2 changes: 2 additions & 0 deletions benchmark/src/Benchmark.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,9 @@ void runSingleThreadTest(std::string configName) {
sketchDimension = cfg->tryU64("sketchDimension", 50, true);
uint64_t coreBind = cfg->tryU64("coreBind", 0, true);
UtilityFunctions::bind2Core((int) coreBind);
torch::set_num_threads(1);
std::string ptFile = cfg->tryString("ptFile", "torchscripts/FDAMM.pt", true);

//uint64_t customResultName = cfg->tryU64("customResultName", 0, true);
INTELLI_INFO("Place me at core" + to_string(coreBind));
INTELLI_INFO(
Expand Down
4 changes: 2 additions & 2 deletions benchmark/torchscripts/CountSketch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@ def CountSketch(A: torch.Tensor, B: torch.Tensor, s: int):
Ca = torch.zeros((m1, s))
Cb = torch.zeros((s, m2))

L = torch.randint(s, (n, ))
G = torch.randint(2, (n, ))
L = torch.randint(s, (n,))
G = torch.randint(2, (n,))

# modify the random column with random sign
for i in range(0, n):
Expand Down
Loading