maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

fix two obvious bugs. (#635)

Browse source
This commit is contained in:
feifeibear 2021-02-16 22:20:03 +08:00 committed by GitHub
parent b901312732
commit 0f5ff94854
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
FasterTransformer/v1/fastertransformer/common.h
View file

@ -14,7 +14,7 @@
* limitations under the License.
*/
#pragma once
#include <stdexcept>
#include <iostream>
#include <cuda_runtime.h>
#include <cuda_fp16.h>

2
FasterTransformer/v1/fastertransformer/trt_plugin/trt_model.h
View file

@ -65,7 +65,7 @@ class TRT_Transformer
auto from_tensor = network->addInput(INPUT_BLOB_NAME, dtype_, nvinfer1::Dims2{seq_len_, hidden_dim_});
auto mask_tensor = network->addInput(MASK_BLOB_NAME, dtype_, nvinfer1::Dims2{seq_len_, seq_len_});
assert(input_tensor);
assert(from_tensor);
assert(mask_tensor);
nvinfer1::ITensor* output_tensor = nullptr;