Skip to content

Commit

Permalink
Disable GPU memory save mode for computing graph, because it's really…
Browse files Browse the repository at this point in the history
… timing-cost.
  • Loading branch information
zhongkaifu committed Feb 16, 2024
1 parent 3ab4c75 commit 8e3e62b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Seq2SeqSharp/Tools/ComputeGraphTensor.cs
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ public ComputeGraphTensor(IWeightFactory weightFactory, int deviceId, bool needB
m_needsBackprop = needBack;
m_deviceId = deviceId;
m_isSubGraph = isSubGraph;
m_saveGPUMemoryMode = saveGPUMemoryMode;
m_saveGPUMemoryMode = false; // We disable it for now, because it's really timing-cost. If you really want to use it, please enable this feature for those specific operators you want to use, such as activate functions.

m_tensorsBindToCurrentGraph = new List<IWeightTensor>();
}
Expand Down

0 comments on commit 8e3e62b

Please sign in to comment.