search

jiangqucheng / torch-mlir-hwacc

Torch-MLIR to HWACC
Other
0 stars 0 forks source link

Register Hwacc GEMM Impl to Execution Engine #4

Open jiangqucheng opened 8 months ago

jiangqucheng commented 8 months ago

Note

The library libEslHwacc.so is dynamically loaded into Python execution environment. Py script first searches the lib location based on the system environment variable LD_LIBRARY_PATH.

https://github.com/jiangqucheng/torch-mlir-hwacc/blob/fa381e6f78bca07ea353e8c03f87a5141e4eb8a1/projects/pt1/python/torch_mlir_e2e_test/hwacc_linalg_on_tensors_backends/hwaccBackendLibCall.py#L78-L103

Therefore, ensure the path of libEslHwacc.so is included in LD_LIBRARY_PATH.

The following is an example before starting python env. 

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$__TORCH_MLIR_PATH__/build/lib