4 ms·

Pretty good perf on AMD compared to CuDNN! What conv kernels are you using? Is PlaidML for inference only?

• hedgehog 9y ago
  Nice! It's all there in the source, they are generated at runtime. Training has some problems with memory consumption but we will fix that.
  • dharma1 9y ago
    Sweet! If you can really get 16bit training going at 26 TFLOPS with Vega 64 then you've done something AMD themselves can't seem to be able to pull off
    • b33pr 9y ago
      Yeah we'll be able to as soon as their OpenCL driver supports it or we write a direct ROCm backend. We have one in the lab now -- we definitely have room to improve its perf. We'll be looking at that a lot more in the future.