diff --git a/kaolin/csrc/ops/spc/convolution.cpp b/kaolin/csrc/ops/spc/convolution.cpp index 668516d1..74559d0d 100644 --- a/kaolin/csrc/ops/spc/convolution.cpp +++ b/kaolin/csrc/ops/spc/convolution.cpp @@ -163,7 +163,7 @@ std::tuple Conv3d_forward( int Qlevel = level; int Plevel = Qlevel - jump; int Olevel = pyramid.size(2)-2; - assert(PLevel >= 0); + assert(Plevel >= 0); uint32_t psize = pyramid.index({ Slice(None), 0, Plevel }).sum().item(); int pmax = pyramid.index({ Slice(None), 0, Plevel }).max().item(); @@ -360,7 +360,7 @@ std::tuple ConvTranspose3d_forward( int Qlevel = level; int Plevel = Qlevel + jump; int Olevel = pyramid.size(2)-2; - assert(PLevel <= Olevel); + assert(Plevel <= Olevel); uint32_t psize = pyramid.index({ Slice(None), 0, Plevel }).sum().item(); int pmax = pyramid.index({ Slice(None), 0, Plevel }).max().item();