Following this video "Accelerate machine learning with Metal" & this sample code "Customizing a TensorFlow operation"
My Environment:
- macOS 13.1
- Python 3.8.15
tensorflow-deps
2.9.0tensorflow-macos
2.9.0tensorflow-metal
0.5.1
after running make
at hash_encoder
, the hash_encoder_kernel.so
appears. Then I tried to run python tiny_nerf_hash.py
but it didn't work.
Unlike the python tiny_nerf_mlp.py
, it works normally but it didn't take the advantage of accelerating.