Disable type inference

2025-11-02 06:24:59 +08:00 · 2025-05-14 13:13:26 -07:00
1 changed files with 1 additions and 0 deletions
--- a/aten/src/ATen/mps/MPSStream.mm
+++ b/aten/src/ATen/mps/MPSStream.mm
@ -29,6 +29,7 @@ MPSStream::MPSStream(Stream stream) : _stream(stream) {

  // Choose level which optimizes for GPU
  _compilationDescriptor.optimizationLevel = MPSGraphOptimizationLevel0;
+  [_compilationDescriptor disableTypeInference];
  _executionDescriptor.compilationDescriptor = _compilationDescriptor;
 }