apply macro at module definition

Signed-off-by: Ceng23333 <441651826@qq.com>

Author: Ceng23333

include/infinicore/nn/embedding.hpp

@@ -75,7 +75,7 @@ class Embedding : public Module {
 
 protected:
     // Parameters
-    Parameter weight_;
+    INFINICORE_NN_PARAMETER(weight);
 
 private:
     size_t num_embeddings_;   // Vocabulary size

include/infinicore/nn/linear.hpp

@@ -31,8 +31,8 @@ class Linear : public Module {
 
 protected:
     // Parameters
-    Parameter weight_;
-    Parameter bias_;
+    INFINICORE_NN_PARAMETER(weight);
+    INFINICORE_NN_PARAMETER(bias);
 
 private:
     // Helper method for common forward computation

include/infinicore/nn/rmsnorm.hpp

@@ -70,7 +70,7 @@ class RMSNorm : public Module {
 
 protected:
     // Parameters
-    Parameter weight_;
+    INFINICORE_NN_PARAMETER(weight);
 
 private:
     size_t normalized_shape_;  // Size of the feature dimension

src/infinicore-test/test_nn_module.h

@@ -21,16 +21,21 @@ namespace infinicore::test {
 // Simple test module that mimics torch.nn.Linear
 class MockLinearModule : public infinicore::nn::Module {
 public:
+    // Declare parameters using macros (torch-like style)
+    INFINICORE_NN_PARAMETER(weight);
+    INFINICORE_NN_PARAMETER(bias);
+
     MockLinearModule(int input_size, int output_size, const infinicore::Device &device)
         : input_size_(input_size), output_size_(output_size), device_(device) {
-
-        // Initialize weight parameter (similar to torch.nn.Linear.weight)
-        register_parameter("weight",
-                           infinicore::nn::Parameter({static_cast<size_t>(output_size), static_cast<size_t>(input_size)}, infinicore::DataType::F32, device));
-
-        // Initialize bias parameter (similar to torch.nn.Linear.bias)
-        register_parameter("bias",
-                           infinicore::nn::Parameter({static_cast<size_t>(output_size)}, infinicore::DataType::F32, device));
+        // Initialize parameters using macros
+        INFINICORE_NN_PARAMETER_INIT(weight,
+                                     ({static_cast<size_t>(output_size), static_cast<size_t>(input_size)},
+                                      infinicore::DataType::F32,
+                                      device));
+        INFINICORE_NN_PARAMETER_INIT(bias,
+                                     ({static_cast<size_t>(output_size)},
+                                      infinicore::DataType::F32,
+                                      device));
     }
 
     // Simple forward pass (conceptual - would need actual matrix operations)