Revert D25434956: [JIT] Use is_buffer in BufferPolicy::valid

Test Plan: revert-hammer Differential Revision: D25434956 (a480ca5) Original commit changeset: ff2229058abb fbshipit-source-id: faba801e9b5e9fa0117624350518592868856eec
pytorch · Dec 12, 2020 · b5b8fe9 · b5b8fe9 · dotcomxtreme1 · Dec 12, 2020
1 parent 693e908
commit b5b8fe9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/csrc/jit/api/module.h b/torch/csrc/jit/api/module.h
@@ -508,7 +508,7 @@ struct TORCH_API BufferPolicy {
   }
   static bool valid(const ClassTypePtr& typ, size_t i, const IValue& v) {
     return typ->getAttribute(i)->isSubtypeOf(TensorType::get()) &&
-        typ->is_buffer(i);
+        !typ->is_parameter(i);
   }
   static CONSTEXPR_EXCEPT_WIN_CUDA bool all_slots = false;
 };