fix a bug in the implementation

39c97588 · Shucai Xiao · b4f6eafa · 39c97588
Commit 39c97588 authored Apr 11, 2019 by Shucai Xiao
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/targets/gpu/fp_conversion.cpp src/targets/gpu/fp_conversion.cpp +1 -0

No files found.
--- a/src/targets/gpu/fp_conversion.cpp
+++ b/src/targets/gpu/fp_conversion.cpp
@@ -9,6 +9,7 @@ namespace gpu {
 shape hip_fp_conversion::compute_shape(std::vector<shape> inputs) const
 {
    inputs.pop_back();
+    check_shapes{inputs}.not_broadcasted().not_transposed();
    return op.compute_shape(inputs);
 }