AidanBeltonS commited on
Commit
9a4e508
·
unverified ·
1 Parent(s): bb0625f

Update get version (llama/6025)

Browse files
Files changed (1) hide show
  1. ggml-sycl.cpp +17 -12
ggml-sycl.cpp CHANGED
@@ -202,24 +202,29 @@ namespace dpct
202
  // Version string has the following format:
203
  // a. OpenCL<space><major.minor><space><vendor-specific-information>
204
  // b. <major.minor>
 
205
  std::string ver;
206
  ver = dev.get_info<sycl::info::device::version>();
207
  std::string::size_type i = 0;
208
- while (i < ver.size())
209
- {
210
- if (isdigit(ver[i]))
211
- break;
212
- i++;
213
  }
214
  major = std::stoi(&(ver[i]));
215
- while (i < ver.size())
216
- {
217
- if (ver[i] == '.')
218
- break;
219
- i++;
 
 
 
 
 
 
 
220
  }
221
- i++;
222
- minor = std::stoi(&(ver[i]));
223
  }
224
 
225
  template <typename tag, typename T>
 
202
  // Version string has the following format:
203
  // a. OpenCL<space><major.minor><space><vendor-specific-information>
204
  // b. <major.minor>
205
+ // c. <AmdGcnArchName> e.g gfx1030
206
  std::string ver;
207
  ver = dev.get_info<sycl::info::device::version>();
208
  std::string::size_type i = 0;
209
+ while (i < ver.size()) {
210
+ if (isdigit(ver[i]))
211
+ break;
212
+ i++;
 
213
  }
214
  major = std::stoi(&(ver[i]));
215
+ while (i < ver.size()) {
216
+ if (ver[i] == '.')
217
+ break;
218
+ i++;
219
+ }
220
+ if (i < ver.size()) {
221
+ // a. and b.
222
+ i++;
223
+ minor = std::stoi(&(ver[i]));
224
+ } else {
225
+ // c.
226
+ minor = 0;
227
  }
 
 
228
  }
229
 
230
  template <typename tag, typename T>