Merge pull request #24350 from benbarsdell:xla-turing-support

PiperOrigin-RevId: 226929463
This commit is contained in:
TensorFlower Gardener 2018-12-26 10:13:12 -08:00
commit 70a9a5cec8

View File

@ -125,6 +125,7 @@ static string GetSmName(std::pair<int, int> compute_capability) {
{{6, 2}, 62},
{{7, 0}, 70},
{{7, 2}, 72},
{{7, 5}, 75},
});
int sm_version = 30;
auto it = m->find(compute_capability);