experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add ARM Mali G68 and G78 GPUs.

Browse Source 
https://developer.arm.com/ip-products/graphics-and-multimedia/mali-gpus/mali-g68-gpu
https://developer.arm.com/ip-products/graphics-and-multimedia/mali-gpus/mali-g78-gpu

PiperOrigin-RevId: 333410430
Change-Id: Icacb4236a8fbd5395d191ee08b8e65d3cdc025ca
This commit is contained in:
Robert David 2020-09-23 17:47:58 -07:00 committed by TensorFlower Gardener
parent 0a9406a47b
commit c680c3a0b3
2 changed files with 6 additions and 2 deletions
tensorflow/lite/delegates/gpu/cl
device_info.ccdevice_info.h

6
tensorflow/lite/delegates/gpu/cl/device_info.cc
View File

@ -40,7 +40,8 @@ MaliGPU GetMaliGPUVersion(const std::string& device_name) {
{"T830", MaliGPU::T830}, {"T860", MaliGPU::T860}, {"T880", MaliGPU::T880},
{"G31", MaliGPU::G31}, {"G51", MaliGPU::G51}, {"G71", MaliGPU::G71},
{"G52", MaliGPU::G52}, {"G72", MaliGPU::G72}, {"G76", MaliGPU::G76},
{"G57", MaliGPU::G57}, {"G77", MaliGPU::G77},
{"G57", MaliGPU::G57}, {"G77", MaliGPU::G77}, {"G68", MaliGPU::G68},
{"G78", MaliGPU::G78},
};
for (const auto& v : kMapping) {
if (device_name.find(v.first) != std::string::npos) {
@ -212,7 +213,8 @@ bool MaliInfo::IsBifrost() const {
}
bool MaliInfo::IsValhall() const {
return gpu_version == MaliGPU::G57 || gpu_version == MaliGPU::G77;
return gpu_version == MaliGPU::G57 || gpu_version == MaliGPU::G77 ||
gpu_version == MaliGPU::G68 || gpu_version == MaliGPU::G78;
}
bool DeviceInfo::SupportsTextureArray() const {

2
tensorflow/lite/delegates/gpu/cl/device_info.h
View File

@ -95,6 +95,8 @@ enum class MaliGPU {
G76,
G57,
G77,
G68,
G78,
UNKNOWN
};