Open ghostplant opened 5 years ago
@ghostplant May I understand your desired goal? Do you wish to run this container rocm/tensorflow:rocm1.9.0-tf1.10-python3
and run all the tensorflow operators on CPU? If that's true, could you try this envrionment variable?
HIP_VISIBLE_DEVICES=-1 /externals/tf_examples/tf_matmul.py
HCC_RUNTIME=CPU
was served for other purposes. To tune what GPUs are visible for HIP applications (such as TensorFlow), please use HIP_VISIBLE_DEVICES
.
Hi @whchung , HIP_VISIBLE_DEVICES=-1
is not my purpose. I want to test how hip data path uses libmcwamp_cpu.so
when AMD GPU (or libmcwamp_hsa.so
) is not available, because I think hcc
was designed for both HSA-based platform and CPU-based platform, right? Such as Fallback to CPU when HSA fails
, etc.
So does this feature work for tensorflow over HIP?
@ghostplant Unfortunately this route (forcing HCC runtime run on CPU and hope HIP applications work) is not supported.
@whchung So do you mean libmcwamp_cpu.so
is actually not useful?
No, libmcwamp_cpu.so
is not useful here. Unfortunately HCC_RUNTIME=CPU
wasn't designed to let HIP applications run on CPU.
@whchung Is there a user example to show what CPU-mode (libmcwamp_cpu.so) is used for? Thanks!
I force the runtime to use CPU mode but tensorflow is not able to launch the computation: