lcy/fix_droplet_runtime_error_202 (#388)

* reset allocatorImpl * restore DIPUBFCachingAllocator.cpp && update deviceimpl.cpp * add comment
DeepLink-org · Nov 6, 2023 · 22457dd · 22457dd
1 parent 8dac466
commit 22457dd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dipu/torch_dipu/csrc_dipu/vendor/droplet/deviceimpl.cpp b/dipu/torch_dipu/csrc_dipu/vendor/droplet/deviceimpl.cpp
@@ -14,7 +14,10 @@ using DROPLET_deviceId = int;
 // =====================
 
 void initializVendor() {
-
+// according to the discussion with droplet team, make a random runtime call 
+// to make sure droplet runtime software is initialized correctly
+  int num = -1;
+  DIPU_CALLDROPLET(::tangGetDeviceCount(reinterpret_cast<int*>(&num)))
 }
 
 void finalizeVendor() {