Remove completed todo

flexflow · Nov 11, 2024 · 12d0fe0 · 12d0fe0
1 parent 99ecd8f
commit 12d0fe0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lib/models/src/models/dlrm/dlrm.cc b/lib/models/src/models/dlrm/dlrm.cc
@@ -132,8 +132,7 @@ ComputationGraph get_dlrm_computation_graph(DLRMConfig const &config) {
 
   tensor_guid_t dense_input = create_input_tensor(
       {config.batch_size, config.dense_arch_layer_sizes.front()},
-      DataType::FLOAT); // TODO: change this to DataType::FLOAT
-                        // after cgb.cast is implemented.
+      DataType::FLOAT);
 
   // Construct the model
   tensor_guid_t bottom_mlp_output = create_dlrm_mlp(