use empty_strided_meta API

NVIDIA · Jan 17, 2025 · 2aad6e6 · 2aad6e6
1 parent c91ff93
commit 2aad6e6
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/csrc/ir/nodes.cpp b/csrc/ir/nodes.cpp
@@ -4537,9 +4537,7 @@ std::vector<PolymorphicValue> MatmulOp::evaluate(
 
   auto matmul_out = at::matmul(a, b);
   const auto& [sizes, strides] = inferShapeOfOutput(out(), ee);
-  auto options =
-      c10::TensorOptions().device(c10::Device(c10::DeviceType::Meta));
-  auto meta_out = at::empty_strided(sizes, strides, options);
+  auto meta_out = at::detail::empty_strided_meta(sizes, strides, a.dtype());
   if (meta_out.is_contiguous()) {
     return {matmul_out};
   }