Skip to content

Instantly share code, notes, and snippets.

This file has been truncated, but you can view the full file.
Requirement already satisfied: transformers[torch] in ./generate-tests/venv/lib/python3.9/site-packages (4.18.0)
Requirement already satisfied: sacremoses in ./generate-tests/venv/lib/python3.9/site-packages (from transformers[torch]) (0.0.49)
Requirement already satisfied: packaging>=20.0 in ./generate-tests/venv/lib/python3.9/site-packages (from transformers[torch]) (21.3)
Requirement already satisfied: pyyaml>=5.1 in ./generate-tests/venv/lib/python3.9/site-packages (from transformers[torch]) (6.0)
Requirement already satisfied: filelock in ./generate-tests/venv/lib/python3.9/site-packages (from transformers[torch]) (3.6.0)
Requirement already satisfied: requests in ./generate-tests/venv/lib/python3.9/site-packages (from transformers[torch]) (2.27.1)
Requirement already satisfied: numpy>=1.17 in ./generate-tests/venv/lib/python3.9/site-packages (from transformers[torch]) (1.22.3)
Requirement already satisfied: huggingface-hub<1.0,>=0.1.0 in ./generate-tests/venv/lib/python3.9/site-packages (from transforme
// -----// IR Dump After FuncBufferize //----- //
module {
func @collapse_dynamic_shape_of_slice(%arg0: memref<?x?x?x?xf32>, %arg1: index, %arg2: index, %arg3: index) -> memref<2x?x?xf32> {
%0 = bufferization.to_tensor %arg0 : memref<?x?x?x?xf32>
%1 = tensor.extract_slice %0[0, 0, %arg1, %arg1] [%arg2, %arg2, %arg3, %arg3] [1, 1, 1, 1] : tensor<?x?x?x?xf32> to tensor<?x?x?x?xf32>
%2 = tensor.cast %1 : tensor<?x?x?x?xf32> to tensor<2x?x?x?xf32>
%3 = tensor.collapse_shape %2 [[0], [1, 2], [3]] : tensor<2x?x?x?xf32> into tensor<2x?x?xf32>
%4 = bufferization.to_memref %3 : memref<2x?x?xf32>
return %4 : memref<2x?x?xf32>
// -----// IR Dump After ConvertLinalgToLLVM //----- //
module {
func @collapse_dynamic_shape(%arg0: memref<2x?x?x?xf32>) -> memref<2x?x?xf32> {
%0 = memref.collapse_shape %arg0 [[0], [1, 2], [3]] : memref<2x?x?x?xf32> into memref<2x?x?xf32>
return %0 : memref<2x?x?xf32>
}
}
ImplicitTypeIDRegistry::lookupOrInsert(mlir::DataLayoutAnalysis)
// -----// IR Dump After ConvertLinalgToLLVM //----- //
module {
func @collapse_dynamic_shape(%arg0: memref<2x?x?x?xf32>) -> memref<2x?x?xf32> {
%0 = memref.collapse_shape %arg0 [[0], [1, 2], [3]] : memref<2x?x?x?xf32> into memref<2x?x?xf32>
return %0 : memref<2x?x?xf32>
}
}
ImplicitTypeIDRegistry::lookupOrInsert(mlir::DataLayoutAnalysis)
"func.func"() ({
^bb0(%arg0: memref<2x?x?x?xf32>):
%0 = "builtin.unrealized_conversion_cast"(%arg0) : (memref<2x?x?x?xf32>) -> !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<4 x i64>, array<4 x i64>)>
%1 = "llvm.mlir.undef"() : () -> !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<3 x i64>, array<3 x i64>)>
%2 = "llvm.extractvalue"(%0) {position = [0]} : (!llvm.struct<(ptr<f32>, ptr<f32>, i64, array<4 x i64>, array<4 x i64>)>) -> !llvm.ptr<f32>
%3 = "llvm.insertvalue"(%1, %2) {position = [0]} : (!llvm.struct<(ptr<f32>, ptr<f32>, i64, array<3 x i64>, array<3 x i64>)>, !llvm.ptr<f32>) -> !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<3 x i64>, array<3 x i64>)>
%4 = "llvm.extractvalue"(%0) {position = [1]} : (!llvm.struct<(ptr<f32>, ptr<f32>, i64, array<4 x i64>, array<4 x i64>)>) -> !llvm.ptr<f32>
%5 = "llvm.insertvalue"(%3, %4) {position = [1]} : (!llvm.struct<(ptr<f32>, ptr<f32>, i64, array<3 x i64>, array<3 x i64>)>, !llvm.ptr<f32>) -> !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<3 x i64>, array<3 x i64
[9/275] ccache /usr/bin/clang++ -DCPUINFO_SUPPORTED_PLATFORM=1 -DIREE_HAL_HAVE_VMVX_DRIVER_MODULE=1 -DIREE_HAL_HAVE_VMVX_MODULE_LOADER=1 -DIREE_HAVE_EMITC_DIALECT -DIREE_HAVE_TORCH_MLIR_DIALECTS -I/usr/local/google/home/cathyzhyi/items/tmtensor/iree/third_party/llvm-project/llvm/include -I/usr/local/google/home/cathyzhyi/items/tmtensor/iree-build/third_party/llvm-project/llvm/include -I/usr/local/google/home/cathyzhyi/items/tmtensor/iree/third_party/llvm-project/mlir/include -I/usr/local/google/home/cathyzhyi/items/tmtensor/iree-build/third_party/llvm-project/llvm/tools/mlir/include -I/usr/local/google/home/cathyzhyi/items/tmtensor/iree-build/third_party/llvm-project/llvm/tools/mlir-hlo/include -I/usr/local/google/home/cathyzhyi/items/tmtensor/iree/third_party/llvm-project/llvm/../mlir/include -I/usr/local/google/home/cathyzhyi/items/tmtensor/iree/llvm-external-projects/iree-dialects/include -I/usr/local/google/home/cathyzhyi/items/tmtensor/iree-build/third_party/llvm-project/llvm/tools/mlir-iree-dialects/inc
slow: 2 processes on CI machine
Ordered by: cumulative time
ncalls tottime percall cumtime percall filename:lineno(function)
1 0.000 0.000 1157.017 1157.017 {built-in method builtins.exec}
404 0.028 0.000 1153.163 2.854 framework.py:283(compile_and_run_test)
399 0.017 0.000 1098.968 2.754 linalg_on_tensors_backend.py:49(run)
400 0.006 0.000 1076.879 2.692 refbackend.py:149(invoke)
400 4.422 0.011 1076.757 2.692 execution_engine.py:26(invoke)
400 1070.587 2.676 1070.595 2.676 execution_engine.py:15(lookup)
slow: 2 processes on CI machine
Ordered by: cumulative time
ncalls tottime percall cumtime percall filename:lineno(function)
1 0.000 0.000 1157.017 1157.017 {built-in method builtins.exec}
404 0.028 0.000 1153.163 2.854 framework.py:283(compile_and_run_test)
399 0.017 0.000 1098.968 2.754 linalg_on_tensors_backend.py:49(run)
400 0.006 0.000 1076.879 2.692 refbackend.py:149(invoke)
400 4.422 0.011 1076.757 2.692 execution_engine.py:26(invoke)
400 1070.587 2.676 1070.595 2.676 execution_engine.py:15(lookup)
slow
Ordered by: cumulative time
ncalls tottime percall cumtime percall filename:lineno(function)
1 0.000 0.000 1157.017 1157.017 {built-in method builtins.exec}
404 0.028 0.000 1153.163 2.854 framework.py:283(compile_and_run_test)
399 0.017 0.000 1098.968 2.754 linalg_on_tensors_backend.py:49(run)
400 0.006 0.000 1076.879 2.692 refbackend.py:149(invoke)
400 4.422 0.011 1076.757 2.692 execution_engine.py:26(invoke)
400 1070.587 2.676 1070.595 2.676 execution_engine.py:15(lookup)
1333685 function calls (1319976 primitive calls) in 1142.703 seconds
Ordered by: cumulative time
ncalls tottime percall cumtime percall filename:lineno(function)
978/1 0.007 0.000 1142.712 1142.712 {built-in method builtins.exec}
1 0.000 0.000 1142.711 1142.711 <string>:1(<module>)
1 0.000 0.000 1142.711 1142.711 runpy.py:200(run_module)
1 0.000 0.000 1142.709 1142.709 runpy.py:64(_run_code)
1 0.000 0.000 1142.708 1142.708 main.py:6(<module>)