Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
comaniac committed Dec 12, 2019
1 parent a521d3a commit c74b003
Show file tree
Hide file tree
Showing 4 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion src/relay/backend/contrib/contrib_codegen.h
Expand Up @@ -212,7 +212,7 @@ class ExternSourcePrinter {
}

/*!
* \briefa A common interface that that used by various external runtime to
* \brief A common interface that is used by various external runtime to
* generate the wrapper to invoke external kernels.
*
* \param subgraph_id The unique id of an external function. It will be used
Expand Down
2 changes: 1 addition & 1 deletion src/relay/backend/contrib/csource/codegen.cc
Expand Up @@ -154,7 +154,7 @@ class CSourceCodegen : public ExternCodegenBase {
code_stream_ << builder.JIT();
}

runtime::Module CreateExternModule(const NodeRef& ref) {
runtime::Module CreateExternModule(const NodeRef& ref) override {
// Create headers
code_stream_ << "#include <cstdint>\n";
code_stream_ << "#include <iostream>\n";
Expand Down
4 changes: 2 additions & 2 deletions src/relay/backend/contrib/dnnl/codegen.cc
Expand Up @@ -229,13 +229,13 @@ class DNNLCodegen : public ExternCodegenBase {
* to compile the generated C source code, users need to specify the paths to
* some libraries, including some TVM required and dnnl specific ones. To make
* linking simpiler, the DNNL kernels are wrapped in a TVM compatible manner
* and are live under include/tvm/runtime/contrib/dnnl folder.
* and live under tvm/src/runtime/contrib/dnnl folder.
*
* \param ref A object ref that could be either a Relay function or module.
*
* \return The runtime module that contains C source code.
*/
runtime::Module CreateExternModule(const NodeRef& ref) {
runtime::Module CreateExternModule(const NodeRef& ref) override {
// Create headers
code_stream_ << "#include <cstdint>\n";
code_stream_ << "#include <cstdlib>\n";
Expand Down
2 changes: 1 addition & 1 deletion tests/python/relay/test_external_codegen.py
Expand Up @@ -177,7 +177,7 @@ def test_extern_dnnl():
if not tvm.get_global_func("relay.contrib.dnnl.enable", True):
print("skip because DNNL codegen is not available")
return

dtype = 'float32'
ishape = (1, 32, 14, 14)
w1shape = (32, 1, 3, 3)
Expand Down

0 comments on commit c74b003

Please sign in to comment.