添加设置推理设备号接口

include/chatglm.h

@@ -42,8 +42,12 @@ namespace fastllm {
 		virtual void SaveLowBitModel(const std::string &fileName, int bit); // 存储成量化模型
 
 		virtual void WarmUp(); // 预热
+
+		int setDeviceID(int id);
     private:
 		virtual void CausalMask(Data &data, int start) {}; // 因果mask？
+
+		int m_device_id=0;
     };
 }