Fix crash on unload torch cpu dll (#67632)

Trying to rebase pytorch/pytorch#61290 into latest pytorch:master Pull Request resolved: pytorch/pytorch#67632 Approved by: https://github.com/ezyang
jjsjann123 · Jul 31, 2022 · a036f0e · a036f0e
1 parent 82c37d3
commit a036f0e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/register_interface.cpp b/register_interface.cpp
@@ -17,7 +17,7 @@ namespace cuda {
 namespace {
 class RegisterInterface {
  public:
- RegisterInterface() {
+ RegisterInterface() : canFuseNodeIndex_(RegisterProfilingNode(canFuseNode)) {
  auto ptr = getFuserInterface();
  ptr->fn_compile_n = &compileCudaFusionGroup;
  ptr->fn_run_n_s = &runCudaFusionGroup;
@@ -27,6 +27,13 @@ class RegisterInterface {
  ptr->fn_profile_n = &shouldProfileNode;
  ptr->fn_skip_n = &skipNodeKind;
  }
+
+ ~RegisterInterface() {
+ DeregisterProfilingNode(canFuseNodeIndex_);
+ }
+
+ private:
+ int canFuseNodeIndex_;
 };
 
 static RegisterInterface register_interface_;