generic.cpp 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747
  1. // Part of the Carbon Language project, under the Apache License v2.0 with LLVM
  2. // Exceptions. See /LICENSE for license information.
  3. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  4. #include "toolchain/check/generic.h"
  5. #include "toolchain/base/kind_switch.h"
  6. #include "toolchain/check/diagnostic_helpers.h"
  7. #include "toolchain/check/eval.h"
  8. #include "toolchain/check/generic_region_stack.h"
  9. #include "toolchain/check/inst.h"
  10. #include "toolchain/check/subst.h"
  11. #include "toolchain/check/type.h"
  12. #include "toolchain/check/type_completion.h"
  13. #include "toolchain/diagnostics/diagnostic.h"
  14. #include "toolchain/sem_ir/constant.h"
  15. #include "toolchain/sem_ir/generic.h"
  16. #include "toolchain/sem_ir/ids.h"
  17. #include "toolchain/sem_ir/inst.h"
  18. #include "toolchain/sem_ir/typed_insts.h"
  19. namespace Carbon::Check {
  20. static auto MakeSelfSpecificId(Context& context, SemIR::GenericId generic_id)
  21. -> SemIR::SpecificId;
  22. // Get the current pending generic. If we have not yet allocated a `GenericId`
  23. // for it, do so now.
  24. static auto GetOrCreatePendingGeneric(Context& context)
  25. -> GenericRegionStack::PendingGeneric {
  26. auto pending_generic = context.generic_region_stack().PeekPendingGeneric();
  27. if (!pending_generic.generic_id.has_value()) {
  28. // Allocate a placeholder generic now to form a generic ID. This generic
  29. // will be populated once we reach the end of the generic declaration.
  30. pending_generic.generic_id = context.generics().Add(
  31. SemIR::Generic{.decl_id = SemIR::InstId::None,
  32. .bindings_id = SemIR::InstBlockId::None,
  33. .self_specific_id = SemIR::SpecificId::None});
  34. context.generic_region_stack().SetPendingGenericId(
  35. pending_generic.generic_id);
  36. }
  37. return pending_generic;
  38. }
  39. // Adds an instruction `generic_inst_id` to the eval block for the current
  40. // generic region. The instruction `generic_inst_id` is expected to compute the
  41. // value of the constant described by `const_inst_id` in each specific. Forms
  42. // and returns a corresponding symbolic constant ID that refers to the
  43. // substituted value of that instruction in each specific.
  44. static auto AddGenericConstantInstToEvalBlock(
  45. Context& context, SemIR::InstId const_inst_id,
  46. SemIR::InstId generic_inst_id, SemIR::ConstantDependence dependence)
  47. -> SemIR::ConstantId {
  48. auto [generic_id, region] = GetOrCreatePendingGeneric(context);
  49. auto index = SemIR::GenericInstIndex(
  50. region, context.generic_region_stack().PeekEvalBlock().size());
  51. context.generic_region_stack().AddInstToEvalBlock(generic_inst_id);
  52. return context.constant_values().AddSymbolicConstant(
  53. {.inst_id = const_inst_id,
  54. .generic_id = generic_id,
  55. .index = index,
  56. .dependence = dependence});
  57. }
  58. namespace {
  59. // Substitution callbacks to rebuild a generic constant in the eval block for a
  60. // generic region.
  61. class RebuildGenericConstantInEvalBlockCallbacks : public SubstInstCallbacks {
  62. public:
  63. // `context` must not be null.
  64. RebuildGenericConstantInEvalBlockCallbacks(Context* context,
  65. SemIR::LocId loc_id)
  66. : SubstInstCallbacks(context),
  67. loc_id_(loc_id),
  68. constants_in_generic_(
  69. context->generic_region_stack().PeekConstantsInGenericMap()) {}
  70. auto RebuildType(SemIR::TypeInstId type_inst_id) const
  71. -> SemIR::TypeId override {
  72. // When building instructions in the eval block, form attached types.
  73. return context().types().GetTypeIdForTypeConstantId(
  74. context().constant_values().GetAttached(type_inst_id));
  75. }
  76. // Check for instructions for which we already have a mapping into the eval
  77. // block, and substitute them with the instructions in the eval block.
  78. auto Subst(SemIR::InstId& inst_id) const -> bool override {
  79. auto const_id = context().constant_values().Get(inst_id);
  80. if (!const_id.has_value()) {
  81. // An unloaded import ref should never contain anything we need to
  82. // substitute into. Don't trigger loading it here.
  83. CARBON_CHECK(
  84. context().insts().Is<SemIR::ImportRefUnloaded>(inst_id),
  85. "Substituting into instruction with invalid constant ID: {0}",
  86. context().insts().Get(inst_id));
  87. return true;
  88. }
  89. if (!context().constant_values().DependsOnGenericParameter(const_id)) {
  90. // This instruction doesn't have a symbolic constant value, so can't
  91. // contain any bindings that need to be substituted.
  92. return true;
  93. }
  94. // If this constant value has a defining instruction in the eval block,
  95. // replace the instruction in the body of the generic with the one from the
  96. // eval block.
  97. if (auto result = constants_in_generic_.Lookup(
  98. context().constant_values().GetInstId(const_id))) {
  99. inst_id = result.value();
  100. return true;
  101. }
  102. return false;
  103. }
  104. // Build a new instruction in the eval block corresponding to the given
  105. // constant.
  106. auto Rebuild(SemIR::InstId orig_inst_id, SemIR::Inst new_inst) const
  107. -> SemIR::InstId override {
  108. auto& orig_symbolic_const = context().constant_values().GetSymbolicConstant(
  109. context().constant_values().Get(orig_inst_id));
  110. auto const_inst_id = orig_symbolic_const.inst_id;
  111. auto dependence = orig_symbolic_const.dependence;
  112. // We might already have an instruction in the eval block if a transitive
  113. // operand of this instruction has the same constant value.
  114. auto result = constants_in_generic_.Insert(const_inst_id, [&] {
  115. // TODO: Add a function on `Context` to add the instruction without
  116. // inserting it into the dependent instructions list or computing a
  117. // constant value for it.
  118. // TODO: Is the location we pick here always appropriate for the new
  119. // instruction?
  120. auto inst_id = context().sem_ir().insts().AddInNoBlock(
  121. SemIR::LocIdAndInst::UncheckedLoc(loc_id_, new_inst));
  122. auto const_id = AddGenericConstantInstToEvalBlock(
  123. context(), const_inst_id, inst_id, dependence);
  124. context().constant_values().Set(inst_id, const_id);
  125. return inst_id;
  126. });
  127. return result.value();
  128. }
  129. auto ReuseUnchanged(SemIR::InstId orig_inst_id) const
  130. -> SemIR::InstId override {
  131. auto inst = context().insts().Get(orig_inst_id);
  132. CARBON_CHECK(
  133. inst.Is<SemIR::BindSymbolicName>() ||
  134. inst.Is<SemIR::SymbolicBindingPattern>(),
  135. "Instruction {0} has symbolic constant value but no symbolic operands",
  136. inst);
  137. // Rebuild the instruction anyway so that it's included in the eval block.
  138. // TODO: Can we just reuse the instruction in this case?
  139. return Rebuild(orig_inst_id, inst);
  140. }
  141. private:
  142. SemIR::LocId loc_id_;
  143. ConstantsInGenericMap& constants_in_generic_;
  144. };
  145. // Substitution callbacks to rebuild a template action. This rebuilds the action
  146. // instruction in-place if it needs to be modified.
  147. class RebuildTemplateActionInEvalBlockCallbacks final
  148. : public RebuildGenericConstantInEvalBlockCallbacks {
  149. public:
  150. // `context` must not be null.
  151. RebuildTemplateActionInEvalBlockCallbacks(Context* context,
  152. SemIR::LocId loc_id,
  153. SemIR::InstId action_inst_id)
  154. : RebuildGenericConstantInEvalBlockCallbacks(context, loc_id),
  155. action_inst_id_(action_inst_id) {}
  156. auto Rebuild(SemIR::InstId orig_inst_id, SemIR::Inst new_inst) const
  157. -> SemIR::InstId override {
  158. if (orig_inst_id == action_inst_id_) {
  159. // TODO: We want to ReplaceInstPreservingConstantValue here, but don't
  160. // want to evaluate the action to check the value hasn't changed.
  161. context().sem_ir().insts().Set(orig_inst_id, new_inst);
  162. return orig_inst_id;
  163. }
  164. return RebuildGenericConstantInEvalBlockCallbacks::Rebuild(orig_inst_id,
  165. new_inst);
  166. }
  167. auto ReuseUnchanged(SemIR::InstId orig_inst_id) const
  168. -> SemIR::InstId override {
  169. if (orig_inst_id == action_inst_id_) {
  170. return orig_inst_id;
  171. }
  172. return RebuildGenericConstantInEvalBlockCallbacks::ReuseUnchanged(
  173. orig_inst_id);
  174. }
  175. private:
  176. SemIR::InstId action_inst_id_;
  177. };
  178. } // namespace
  179. // Adds instructions to compute the substituted version of `type_id` in each
  180. // specific into the eval block for the current generic region. Returns a
  181. // symbolic type ID that refers to the substituted type in each specific.
  182. static auto AddGenericTypeToEvalBlock(Context& context, SemIR::LocId loc_id,
  183. SemIR::TypeId type_id) -> SemIR::TypeId {
  184. // Substitute into the type's constant instruction and rebuild it in the eval
  185. // block.
  186. auto type_inst_id =
  187. SubstInst(context, context.types().GetInstId(type_id),
  188. RebuildGenericConstantInEvalBlockCallbacks(&context, loc_id));
  189. return context.types().GetTypeIdForTypeConstantId(
  190. context.constant_values().GetAttached(type_inst_id));
  191. }
  192. // Adds instructions to compute the substituted value of `inst_id` in each
  193. // specific into the eval block for the current generic region. Returns a
  194. // symbolic constant instruction ID that refers to the substituted constant
  195. // value in each specific.
  196. static auto AddGenericConstantToEvalBlock(Context& context,
  197. SemIR::InstId inst_id)
  198. -> SemIR::ConstantId {
  199. CARBON_CHECK(context.constant_values().Get(inst_id).is_symbolic(),
  200. "Adding generic constant {0} with non-symbolic value {1}",
  201. context.insts().Get(inst_id),
  202. context.constant_values().Get(inst_id));
  203. // Substitute into the constant value and rebuild it in the eval block if
  204. // we've not encountered it before.
  205. auto const_inst_id = context.constant_values().GetConstantInstId(inst_id);
  206. auto callbacks = RebuildGenericConstantInEvalBlockCallbacks(
  207. &context, SemIR::LocId(inst_id));
  208. auto new_inst_id = SubstInst(context, const_inst_id, callbacks);
  209. CARBON_CHECK(new_inst_id != const_inst_id,
  210. "No substitutions performed for generic constant {0}",
  211. context.insts().Get(inst_id));
  212. return context.constant_values().GetAttached(new_inst_id);
  213. }
  214. // Adds an instruction that performs a template action to the eval block for the
  215. // generic. The instruction should not yet have been added to any block. The
  216. // instruction might refer to types and constants that need to be rewritten, so
  217. // substitute into it first.
  218. static auto AddTemplateActionToEvalBlock(Context& context,
  219. SemIR::InstId inst_id) -> void {
  220. // Substitute into the constant value and rebuild it in the eval block.
  221. auto new_inst_id = SubstInst(context, inst_id,
  222. RebuildTemplateActionInEvalBlockCallbacks(
  223. &context, SemIR::LocId(inst_id), inst_id));
  224. CARBON_CHECK(new_inst_id == inst_id,
  225. "Substitution changed InstId of template action");
  226. context.generic_region_stack().PeekConstantsInGenericMap().Insert(inst_id,
  227. inst_id);
  228. // Add the action to the eval block and point its constant value back to its
  229. // index within the block.
  230. auto [generic_id, region] = GetOrCreatePendingGeneric(context);
  231. auto& symbolic_constant = context.constant_values().GetSymbolicConstant(
  232. context.constant_values().GetAttached(inst_id));
  233. symbolic_constant.generic_id = generic_id;
  234. symbolic_constant.index = SemIR::GenericInstIndex(
  235. region, context.generic_region_stack().PeekEvalBlock().size());
  236. context.generic_region_stack().AddInstToEvalBlock(inst_id);
  237. }
  238. // Populates a map of constants in a generic from the constants in the
  239. // declaration region, in preparation for building the definition region.
  240. static auto PopulateConstantsFromDeclaration(
  241. Context& context, SemIR::GenericId generic_id,
  242. ConstantsInGenericMap& constants_in_generic) {
  243. // For the definition region, populate constants from the declaration.
  244. auto decl_eval_block = context.inst_blocks().Get(
  245. context.generics().Get(generic_id).decl_block_id);
  246. constants_in_generic.GrowForInsertCount(decl_eval_block.size());
  247. for (auto inst_id : decl_eval_block) {
  248. auto const_inst_id = context.constant_values().GetConstantInstId(inst_id);
  249. auto result = constants_in_generic.Insert(const_inst_id, inst_id);
  250. CARBON_CHECK(result.is_inserted(),
  251. "Duplicate constant in generic decl eval block: {0}",
  252. context.insts().Get(const_inst_id));
  253. }
  254. }
  255. auto AttachDependentInstToCurrentGeneric(Context& context,
  256. DependentInst dependent_inst) -> void {
  257. auto [inst_id, dep_kind] = dependent_inst;
  258. // If we don't have a generic region here, leave the dependent instruction
  259. // unattached. This happens for out-of-line redeclarations of members of
  260. // dependent scopes:
  261. //
  262. // class A(T:! type) {
  263. // fn F();
  264. // }
  265. // // Has generic type and constant value, but no generic region.
  266. // fn A(T:! type).F() {}
  267. //
  268. // TODO: Copy the attached type and constant value from the previous
  269. // declaration in this case instead of attempting to attach the new
  270. // declaration to a generic region that we're no longer within.
  271. if (context.generic_region_stack().Empty()) {
  272. // This should only happen for `*Decl` instructions, never for template
  273. // actions.
  274. CARBON_CHECK((dep_kind & DependentInst::Template) == DependentInst::None);
  275. return;
  276. }
  277. context.generic_region_stack().AddDependentInst(dependent_inst.inst_id);
  278. // If the type is symbolic, replace it with a type specific to this generic.
  279. if ((dep_kind & DependentInst::SymbolicType) != DependentInst::None) {
  280. auto inst = context.insts().Get(inst_id);
  281. auto type_id = AddGenericTypeToEvalBlock(context, SemIR::LocId(inst_id),
  282. inst.type_id());
  283. // TODO: Eventually, completeness requirements should be modeled as
  284. // constraints on the generic rather than properties of the type. For now,
  285. // require the transformed type to be complete if the original was.
  286. if (context.types().IsComplete(inst.type_id())) {
  287. CompleteTypeOrCheckFail(context, type_id);
  288. }
  289. inst.SetType(type_id);
  290. context.sem_ir().insts().Set(inst_id, inst);
  291. }
  292. // If the instruction has a symbolic constant value, then make a note that
  293. // we'll need to evaluate this instruction when forming the specific. Update
  294. // the constant value of the instruction to refer to the result of that
  295. // eventual evaluation.
  296. if ((dep_kind & DependentInst::SymbolicConstant) != DependentInst::None) {
  297. // Update the constant value to refer to this generic.
  298. context.constant_values().Set(
  299. inst_id, AddGenericConstantToEvalBlock(context, inst_id));
  300. }
  301. // If the instruction is a template action, add it directly to this position
  302. // in the eval block.
  303. if ((dep_kind & DependentInst::Template) != DependentInst::None) {
  304. AddTemplateActionToEvalBlock(context, inst_id);
  305. }
  306. }
  307. // Builds and returns a block of instructions whose constant values need to be
  308. // evaluated in order to resolve a generic to a specific.
  309. static auto MakeGenericEvalBlock(Context& context) -> SemIR::InstBlockId {
  310. return context.inst_blocks().Add(
  311. context.generic_region_stack().PeekEvalBlock());
  312. }
  313. // Builds and returns an eval block, given the list of canonical symbolic
  314. // constants that the instructions in the eval block should produce. This is
  315. // used when importing a generic.
  316. auto RebuildGenericEvalBlock(Context& context, SemIR::GenericId generic_id,
  317. SemIR::GenericInstIndex::Region region,
  318. llvm::ArrayRef<SemIR::InstId> const_ids)
  319. -> SemIR::InstBlockId {
  320. context.generic_region_stack().Push(
  321. {.generic_id = generic_id, .region = region});
  322. auto& constants_in_generic =
  323. context.generic_region_stack().PeekConstantsInGenericMap();
  324. // For the definition region, populate constants from the declaration.
  325. if (region == SemIR::GenericInstIndex::Definition) {
  326. PopulateConstantsFromDeclaration(context, generic_id, constants_in_generic);
  327. }
  328. constants_in_generic.GrowForInsertCount(const_ids.size());
  329. for (auto [i, inst_id] : llvm::enumerate(const_ids)) {
  330. // Build a constant in the inst block.
  331. AddGenericConstantToEvalBlock(context, inst_id);
  332. CARBON_CHECK(context.generic_region_stack().PeekEvalBlock().size() == i + 1,
  333. "Produced {0} instructions when importing {1}",
  334. (context.generic_region_stack().PeekEvalBlock().size() - i),
  335. context.insts().Get(inst_id));
  336. }
  337. auto eval_block_id = MakeGenericEvalBlock(context);
  338. context.generic_region_stack().Pop();
  339. return eval_block_id;
  340. }
  341. auto StartGenericDecl(Context& context) -> void {
  342. context.generic_region_stack().Push(
  343. {.generic_id = SemIR::GenericId::None,
  344. .region = SemIR::GenericInstIndex::Declaration});
  345. }
  346. auto StartGenericDefinition(Context& context, SemIR::GenericId generic_id)
  347. -> void {
  348. // Push a generic region even if we don't have a generic_id. We might still
  349. // have locally-introduced generic parameters to track:
  350. //
  351. // fn F() {
  352. // let T:! type = i32;
  353. // var x: T;
  354. // }
  355. context.generic_region_stack().Push(
  356. {.generic_id = generic_id,
  357. .region = SemIR::GenericInstIndex::Definition});
  358. if (generic_id.has_value()) {
  359. PopulateConstantsFromDeclaration(
  360. context, generic_id,
  361. context.generic_region_stack().PeekConstantsInGenericMap());
  362. }
  363. }
  364. auto DiscardGenericDecl(Context& context) -> void {
  365. // Unattach any types and constant values we might have created in the
  366. // generic.
  367. // TODO: We should re-evaluate the contents of the eval block in a synthesized
  368. // specific to form these values, in order to propagate the values of local
  369. // `let :!` bindings.
  370. for (auto inst_id : context.generic_region_stack().PeekDependentInsts()) {
  371. // Note that `Get` returns an instruction with an unattached type.
  372. context.sem_ir().insts().Set(inst_id, context.insts().Get(inst_id));
  373. // Note that `Get` returns an unattached constant.
  374. context.constant_values().Set(inst_id,
  375. context.constant_values().Get(inst_id));
  376. }
  377. // Note that we may leak a GenericId here, if one was allocated.
  378. context.generic_region_stack().Pop();
  379. }
  380. auto BuildGeneric(Context& context, SemIR::InstId decl_id) -> SemIR::GenericId {
  381. auto all_bindings =
  382. context.scope_stack().compile_time_bindings_stack().PeekAllValues();
  383. if (all_bindings.empty()) {
  384. CARBON_CHECK(context.generic_region_stack().PeekEvalBlock().empty(),
  385. "Have non-empty eval block {0} in declaration {1} but no "
  386. "compile time bindings are in scope.",
  387. context.insts().Get(
  388. context.generic_region_stack().PeekEvalBlock().front()),
  389. context.insts().Get(decl_id));
  390. DiscardGenericDecl(context);
  391. return SemIR::GenericId::None;
  392. }
  393. // Build the new Generic object. Note that we intentionally do not hold a
  394. // persistent reference to it throughout this function, because the `generics`
  395. // collection can have items added to it by import resolution while we are
  396. // building this generic.
  397. auto bindings_id = context.inst_blocks().Add(all_bindings);
  398. SemIR::Generic generic = {.decl_id = decl_id,
  399. .bindings_id = bindings_id,
  400. .self_specific_id = SemIR::SpecificId::None};
  401. // Get the generic ID, or allocate one now if we don't have one yet. That
  402. // could happen if the eval block is empty.
  403. auto generic_id =
  404. context.generic_region_stack().PeekPendingGeneric().generic_id;
  405. if (!generic_id.has_value()) {
  406. CARBON_CHECK(context.generic_region_stack().PeekEvalBlock().empty(),
  407. "Non-empty eval block but didn't yet allocate a GenericId");
  408. generic_id = context.generics().Add(generic);
  409. context.generic_region_stack().SetPendingGenericId(generic_id);
  410. } else {
  411. CARBON_CHECK(!context.generics().Get(generic_id).decl_id.has_value(),
  412. "Built generic {0} twice", generic_id);
  413. context.generics().Get(generic_id) = generic;
  414. }
  415. // MakeSelfSpecificId could cause something to be imported, which would
  416. // invalidate the return value of `context.generics().Get(generic_id)`.
  417. auto self_specific_id = MakeSelfSpecificId(context, generic_id);
  418. context.generics().Get(generic_id).self_specific_id = self_specific_id;
  419. return generic_id;
  420. }
  421. auto FinishGenericDecl(Context& context, SemIR::LocId loc_id,
  422. SemIR::GenericId generic_id) -> void {
  423. if (!generic_id.has_value()) {
  424. return;
  425. }
  426. auto decl_block_id = MakeGenericEvalBlock(context);
  427. context.generic_region_stack().Pop();
  428. context.generics().Get(generic_id).decl_block_id = decl_block_id;
  429. ResolveSpecificDeclaration(context, loc_id,
  430. context.generics().GetSelfSpecific(generic_id));
  431. }
  432. auto BuildGenericDecl(Context& context, SemIR::InstId decl_id)
  433. -> SemIR::GenericId {
  434. SemIR::GenericId generic_id = BuildGeneric(context, decl_id);
  435. if (generic_id.has_value()) {
  436. FinishGenericDecl(context, SemIR::LocId(decl_id), generic_id);
  437. }
  438. return generic_id;
  439. }
  440. // Returns the first difference between the two given eval blocks.
  441. static auto FirstDifferenceBetweenEvalBlocks(
  442. Context& context, llvm::ArrayRef<SemIR::InstId> old_eval_block,
  443. llvm::ArrayRef<SemIR::InstId> new_eval_block)
  444. -> std::pair<SemIR::InstId, SemIR::InstId> {
  445. // Check each element of the eval block computes the same unattached constant.
  446. for (auto [old_inst_id, new_inst_id] :
  447. llvm::zip(old_eval_block, new_eval_block)) {
  448. auto old_const_id = context.constant_values().Get(old_inst_id);
  449. auto new_const_id = context.constant_values().Get(new_inst_id);
  450. if (old_const_id != new_const_id) {
  451. if (old_const_id.is_symbolic() && new_const_id.is_symbolic() &&
  452. context.constant_values().GetDependence(old_const_id) ==
  453. SemIR::ConstantDependence::Template &&
  454. context.constant_values().GetDependence(new_const_id) ==
  455. SemIR::ConstantDependence::Template &&
  456. context.insts().Get(old_inst_id).kind() ==
  457. context.insts().Get(new_inst_id).kind()) {
  458. // TODO: We don't have a good mechanism to compare template constants
  459. // because they canonicalize to themselves, so just assume this is OK.
  460. continue;
  461. }
  462. // These constant values differ unexpectedly.
  463. return {old_inst_id, new_inst_id};
  464. }
  465. }
  466. if (old_eval_block.size() < new_eval_block.size()) {
  467. return {SemIR::InstId::None, new_eval_block[old_eval_block.size()]};
  468. }
  469. if (old_eval_block.size() > new_eval_block.size()) {
  470. return {old_eval_block[new_eval_block.size()], SemIR::InstId::None};
  471. }
  472. return {SemIR::InstId::None, SemIR::InstId::None};
  473. }
  474. // If `constant_id` refers to a symbolic constant within the declaration region
  475. // of `generic_id`, remap it to refer to the constant value of the corresponding
  476. // element in the given eval block. Otherwise returns the ID unchanged.
  477. static auto ReattachConstant(Context& context, SemIR::GenericId generic_id,
  478. llvm::ArrayRef<SemIR::InstId> eval_block,
  479. SemIR::ConstantId constant_id)
  480. -> SemIR::ConstantId {
  481. if (!constant_id.has_value() || !constant_id.is_symbolic()) {
  482. return constant_id;
  483. }
  484. auto& symbolic_const =
  485. context.constant_values().GetSymbolicConstant(constant_id);
  486. if (symbolic_const.generic_id != generic_id) {
  487. // Constant doesn't refer into this generic.
  488. return constant_id;
  489. }
  490. CARBON_CHECK(
  491. symbolic_const.index.region() == SemIR::GenericInstIndex::Declaration,
  492. "Definition region of redeclaration should not be referenced");
  493. return context.constant_values().GetAttached(
  494. eval_block[symbolic_const.index.index()]);
  495. }
  496. // Same as `ReattachConstant` but for a type.
  497. static auto ReattachType(Context& context, SemIR::GenericId generic_id,
  498. llvm::ArrayRef<SemIR::InstId> eval_block,
  499. SemIR::TypeId type_id) -> SemIR::TypeId {
  500. return context.types().GetTypeIdForTypeConstantId(ReattachConstant(
  501. context, generic_id, eval_block, context.types().GetConstantId(type_id)));
  502. }
  503. auto FinishGenericRedecl(Context& context, SemIR::GenericId generic_id)
  504. -> void {
  505. if (!generic_id.has_value()) {
  506. DiscardGenericDecl(context);
  507. return;
  508. }
  509. // Find the old and new eval blocks.
  510. auto old_eval_block_id =
  511. context.generics()
  512. .Get(generic_id)
  513. .GetEvalBlock(SemIR::GenericInstIndex::Declaration);
  514. CARBON_CHECK(old_eval_block_id.has_value(),
  515. "Old generic is not fully declared");
  516. auto old_eval_block = context.inst_blocks().Get(old_eval_block_id);
  517. auto new_eval_block = context.generic_region_stack().PeekEvalBlock();
  518. // Check the eval blocks are computing the same constants in the same order.
  519. // This should always be the case because we have already verified they have
  520. // the same parse tree, and the poisoning rules mean that all entities they
  521. // refer to are also the same.
  522. //
  523. // Note that it's OK if the first difference is that an old instruction has no
  524. // corresponding new instruction; we wouldn't have used that anyway. This
  525. // happens for `ImplDecl`, for which the witness is included in the eval block
  526. // of the first declaration.
  527. if (auto [old_inst_id, new_inst_id] = FirstDifferenceBetweenEvalBlocks(
  528. context, old_eval_block, new_eval_block);
  529. new_inst_id.has_value()) {
  530. // This shouldn't be possible: we should have already checked that the
  531. // syntax of the redeclaration matches the prior declaration, and none of
  532. // the name lookups or semantic checks should be allowed to differ between
  533. // the two declarations, so we should have built the same eval block as in
  534. // the prior declaration.
  535. //
  536. // However, that isn't a strong enough invariant that it seems appropriate
  537. // to CHECK-fail here, so we produce a diagnostic with context.TODO()
  538. // instead.
  539. //
  540. // TODO: Add something like context.UNEXPECTED() instead of using
  541. // context.TODO() here because there's not really anything to do.
  542. context.TODO(new_inst_id,
  543. "generic redeclaration differs from previous declaration");
  544. if (old_inst_id.has_value()) {
  545. context.TODO(old_inst_id, "instruction in previous declaration");
  546. }
  547. DiscardGenericDecl(context);
  548. return;
  549. }
  550. auto redecl_generic_id =
  551. context.generic_region_stack().PeekPendingGeneric().generic_id;
  552. // Reattach any instructions that depend on the redeclaration to instead refer
  553. // to the original.
  554. for (auto inst_id : context.generic_region_stack().PeekDependentInsts()) {
  555. // Reattach the type.
  556. auto inst = context.insts().GetWithAttachedType(inst_id);
  557. inst.SetType(ReattachType(context, redecl_generic_id, old_eval_block,
  558. inst.type_id()));
  559. context.sem_ir().insts().Set(inst_id, inst);
  560. // Reattach the constant value.
  561. context.constant_values().Set(
  562. inst_id,
  563. ReattachConstant(context, redecl_generic_id, old_eval_block,
  564. context.constant_values().GetAttached(inst_id)));
  565. }
  566. context.generic_region_stack().Pop();
  567. }
  568. auto FinishGenericDefinition(Context& context, SemIR::GenericId generic_id)
  569. -> void {
  570. if (!generic_id.has_value()) {
  571. DiscardGenericDecl(context);
  572. return;
  573. }
  574. auto definition_block_id = MakeGenericEvalBlock(context);
  575. context.generic_region_stack().Pop();
  576. context.generics().Get(generic_id).definition_block_id = definition_block_id;
  577. }
  578. auto ResolveSpecificDeclaration(Context& context, SemIR::LocId loc_id,
  579. SemIR::SpecificId specific_id) -> void {
  580. // If this is the first time we've formed this specific, evaluate its decl
  581. // block to form information about the specific.
  582. if (!context.specifics().Get(specific_id).decl_block_id.has_value()) {
  583. // Set a placeholder value as the decl block ID so we won't attempt to
  584. // recursively resolve the same specific.
  585. context.specifics().Get(specific_id).decl_block_id =
  586. SemIR::InstBlockId::Empty;
  587. auto decl_block_id =
  588. TryEvalBlockForSpecific(context, loc_id, specific_id,
  589. SemIR::GenericInstIndex::Region::Declaration);
  590. // Note that TryEvalBlockForSpecific may reallocate the list of specifics,
  591. // so re-lookup the specific here.
  592. context.specifics().Get(specific_id).decl_block_id = decl_block_id;
  593. }
  594. }
  595. auto MakeSpecific(Context& context, SemIR::LocId loc_id,
  596. SemIR::GenericId generic_id, SemIR::InstBlockId args_id)
  597. -> SemIR::SpecificId {
  598. auto specific_id = context.specifics().GetOrAdd(generic_id, args_id);
  599. ResolveSpecificDeclaration(context, loc_id, specific_id);
  600. return specific_id;
  601. }
  602. auto MakeSpecific(Context& context, SemIR::LocId loc_id,
  603. SemIR::GenericId generic_id,
  604. llvm::ArrayRef<SemIR::InstId> args) -> SemIR::SpecificId {
  605. auto args_id = context.inst_blocks().AddCanonical(args);
  606. return MakeSpecific(context, loc_id, generic_id, args_id);
  607. }
  608. static auto MakeSelfSpecificId(Context& context, SemIR::GenericId generic_id)
  609. -> SemIR::SpecificId {
  610. if (!generic_id.has_value()) {
  611. return SemIR::SpecificId::None;
  612. }
  613. auto& generic = context.generics().Get(generic_id);
  614. auto args = context.inst_blocks().Get(generic.bindings_id);
  615. // Form a canonical argument list for the generic.
  616. llvm::SmallVector<SemIR::InstId> arg_ids;
  617. arg_ids.reserve(args.size());
  618. for (auto arg_id : args) {
  619. arg_ids.push_back(context.constant_values().GetConstantInstId(arg_id));
  620. }
  621. auto args_id = context.inst_blocks().AddCanonical(arg_ids);
  622. return context.specifics().GetOrAdd(generic_id, args_id);
  623. }
  624. auto MakeSelfSpecific(Context& context, SemIR::LocId loc_id,
  625. SemIR::GenericId generic_id) -> SemIR::SpecificId {
  626. // Build a corresponding specific.
  627. SemIR::SpecificId specific_id = MakeSelfSpecificId(context, generic_id);
  628. // TODO: This could be made more efficient. We don't need to perform
  629. // substitution here; we know we want identity mappings for all constants and
  630. // types. We could also consider not storing the mapping at all in this case.
  631. ResolveSpecificDeclaration(context, loc_id, specific_id);
  632. return specific_id;
  633. }
  634. auto ResolveSpecificDefinition(Context& context, SemIR::LocId loc_id,
  635. SemIR::SpecificId specific_id) -> bool {
  636. // TODO: Handle recursive resolution of the same generic definition.
  637. auto& specific = context.specifics().Get(specific_id);
  638. auto generic_id = specific.generic_id;
  639. CARBON_CHECK(generic_id.has_value(), "Specific with no generic ID");
  640. if (!specific.definition_block_id.has_value()) {
  641. // Evaluate the eval block for the definition of the generic.
  642. auto& generic = context.generics().Get(generic_id);
  643. if (!generic.definition_block_id.has_value()) {
  644. // The generic is not defined yet.
  645. return false;
  646. }
  647. auto definition_block_id = TryEvalBlockForSpecific(
  648. context, loc_id, specific_id, SemIR::GenericInstIndex::Definition);
  649. // Note that TryEvalBlockForSpecific may reallocate the list of specifics,
  650. // so re-lookup the specific here.
  651. context.specifics().Get(specific_id).definition_block_id =
  652. definition_block_id;
  653. }
  654. return true;
  655. }
  656. auto DiagnoseIfGenericMissingExplicitParameters(
  657. Context& context, SemIR::EntityWithParamsBase& entity_base) -> void {
  658. if (!entity_base.implicit_param_patterns_id.has_value() ||
  659. entity_base.param_patterns_id.has_value()) {
  660. return;
  661. }
  662. CARBON_DIAGNOSTIC(GenericMissingExplicitParameters, Error,
  663. "expected explicit parameters after implicit parameters");
  664. context.emitter().Emit(entity_base.last_param_node_id,
  665. GenericMissingExplicitParameters);
  666. }
  667. } // namespace Carbon::Check