diff --git a/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/ivalue_importer.cpp b/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/ivalue_importer.cpp index 00c0dcc10470..cea471c3d68d 100644 --- a/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/ivalue_importer.cpp +++ b/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/ivalue_importer.cpp @@ -21,7 +21,7 @@ #include "mlir-c/Diagnostics.h" #include "torch-mlir-c/TorchTypes.h" -#include "ATen/native/quantized/cpu/PackedParams.h" +#include "ATen/native/quantized/PackedParams.h" #include "caffe2/core/scope_guard.h" using namespace torch_mlir; diff --git a/requirements.txt b/requirements.txt index dab7bce187b4..5bd48f90c058 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ --f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html +-f https://download.pytorch.org/whl/nightly/cpu --pre torch