buildscript

xcrun -sdk macosx metal -o flash.ir  -c flash.metal;
xcrun -sdk macosx metallib -o flash.metallib flash.ir;

xcrun -sdk macosx metal -o flashback.ir  -c flashback.metal;
xcrun -sdk macosx metallib -o flashback.metallib flashback.ir;

g++ -std=c++17 -O3 -shared -fPIC -DTORCH_EXTENSION_NAME=FlashMetal $(python -m pybind11 --includes) \
driver.cpp \
-I/opt/homebrew/Caskroom/miniforge/base/envs/mlp/lib/python3.8/site-packages/torch/include \
-I/opt/homebrew/Caskroom/miniforge/base/envs/mlp/lib/python3.8/site-packages/torch/include/torch/csrc/api/include \
-I ~/metal-cpp \
-L/opt/homebrew/Caskroom/miniforge/base/envs/mlp/lib/python3.8/site-packages/torch/lib \
-Wl,-rpath,/opt/homebrew/Caskroom/miniforge/base/envs/mlp/lib/python3.8/site-packages/torch/lib \
-lc10 -ltorch -ltorch_cpu -framework Foundation -framework QuartzCore -framework Metal \
-Wl,-undefined,dynamic_lookup \
-o FlashMetal`python3-config --extension-suffix`;

rm *.ir