update cuDLA samples

fix missing DPRINTF
fix Makefile
This commit is contained in:
Rutwik Choughule 2021-11-23 14:34:52 +05:30
parent e64c65a0d3
commit 3a05f29b94
8 changed files with 15 additions and 4 deletions

View File

@ -1,5 +1,5 @@
################################################################################ ################################################################################
# Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved. # Copyright (c) 2021, NVIDIA CORPORATION. All rights reserved.
# #
# Redistribution and use in source and binary forms, with or without # Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions # modification, are permitted provided that the following conditions
@ -346,6 +346,8 @@ endif
ALL_CCFLAGS += --std=c++11 --threads 0 ALL_CCFLAGS += --std=c++11 --threads 0
LIBRARIES += -lcudla
ifeq ($(SAMPLE_ENABLED),0) ifeq ($(SAMPLE_ENABLED),0)
EXEC ?= @echo "[@]" EXEC ?= @echo "[@]"
endif endif

View File

@ -22,6 +22,7 @@
<keyword>CPP11</keyword> <keyword>CPP11</keyword>
</keywords> </keywords>
<libraries> <libraries>
<library>cudla</library>
</libraries> </libraries>
<librarypaths> <librarypaths>
</librarypaths> </librarypaths>

View File

@ -35,6 +35,8 @@
#include <fstream> #include <fstream>
#include <sstream> #include <sstream>
#define DPRINTF(...) printf(__VA_ARGS__)
static void printTensorDesc(cudlaModuleTensorDescriptor* tensorDesc) { static void printTensorDesc(cudlaModuleTensorDescriptor* tensorDesc) {
DPRINTF("\tTENSOR NAME : %s\n", tensorDesc->name); DPRINTF("\tTENSOR NAME : %s\n", tensorDesc->name);
DPRINTF("\tsize: %lu\n", tensorDesc->size); DPRINTF("\tsize: %lu\n", tensorDesc->size);

View File

@ -1,5 +1,5 @@
################################################################################ ################################################################################
# Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved. # Copyright (c) 2021, NVIDIA CORPORATION. All rights reserved.
# #
# Redistribution and use in source and binary forms, with or without # Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions # modification, are permitted provided that the following conditions
@ -346,6 +346,8 @@ endif
ALL_CCFLAGS += --std=c++11 --threads 0 ALL_CCFLAGS += --std=c++11 --threads 0
LIBRARIES += -lcudla
ifeq ($(SAMPLE_ENABLED),0) ifeq ($(SAMPLE_ENABLED),0)
EXEC ?= @echo "[@]" EXEC ?= @echo "[@]"
endif endif

View File

@ -22,6 +22,7 @@
<keyword>CPP11</keyword> <keyword>CPP11</keyword>
</keywords> </keywords>
<libraries> <libraries>
<library>cudla</library>
</libraries> </libraries>
<librarypaths> <librarypaths>
</librarypaths> </librarypaths>

View File

@ -1,5 +1,5 @@
################################################################################ ################################################################################
# Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved. # Copyright (c) 2021, NVIDIA CORPORATION. All rights reserved.
# #
# Redistribution and use in source and binary forms, with or without # Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions # modification, are permitted provided that the following conditions
@ -349,7 +349,7 @@ endif
ALL_CCFLAGS += --std=c++11 --threads 0 ALL_CCFLAGS += --std=c++11 --threads 0
LIBRARIES += -lnvscibuf -lnvscisync LIBRARIES += -lcudla -lnvscibuf -lnvscisync
ifeq ($(SAMPLE_ENABLED),0) ifeq ($(SAMPLE_ENABLED),0)
EXEC ?= @echo "[@]" EXEC ?= @echo "[@]"

View File

@ -22,6 +22,7 @@
<keyword>CPP11</keyword> <keyword>CPP11</keyword>
</keywords> </keywords>
<libraries> <libraries>
<library>cudla</library>
<library>nvscibuf</library> <library>nvscibuf</library>
<library>nvscisync</library> <library>nvscisync</library>
</libraries> </libraries>

View File

@ -37,6 +37,8 @@
#include <sstream> #include <sstream>
#include <unistd.h> #include <unistd.h>
#define DPRINTF(...) printf(__VA_ARGS__)
static void printTensorDesc(cudlaModuleTensorDescriptor* tensorDesc) { static void printTensorDesc(cudlaModuleTensorDescriptor* tensorDesc) {
DPRINTF("\tTENSOR NAME : %s\n", tensorDesc->name); DPRINTF("\tTENSOR NAME : %s\n", tensorDesc->name);
DPRINTF("\tsize: %lu\n", tensorDesc->size); DPRINTF("\tsize: %lu\n", tensorDesc->size);