From d6f0530215660d122582c800fafade339172ff1c Mon Sep 17 00:00:00 2001
From: Ivan Kylchik <ivan.kylchik@jetbrains.com>
Date: Tue, 24 Oct 2023 18:24:08 +0200
Subject: [PATCH] [JVM] Use `getContextContainingFile` method in
 `CodegenBinding`

---
 .../jetbrains/kotlin/codegen/binding/CodegenBinding.java | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/compiler/backend/src/org/jetbrains/kotlin/codegen/binding/CodegenBinding.java b/compiler/backend/src/org/jetbrains/kotlin/codegen/binding/CodegenBinding.java
index 68026774363..88709b7f674 100644
--- a/compiler/backend/src/org/jetbrains/kotlin/codegen/binding/CodegenBinding.java
+++ b/compiler/backend/src/org/jetbrains/kotlin/codegen/binding/CodegenBinding.java
@@ -99,12 +99,9 @@ public class CodegenBinding {
         for (KtFile file : allFilesInPackages(state.getBindingContext(), files)) {
             file.accept(visitor);
             if (file instanceof KtCodeFragment) {
-                PsiElement context = file.getContext();
-                if (context != null) {
-                    PsiFile contextFile = context.getContainingFile();
-                    if (contextFile != null) {
-                        contextFile.accept(visitor);
-                    }
+                PsiFile contextFile = ((KtCodeFragment) file).getContextContainingFile();
+                if (contextFile != null) {
+                    contextFile.accept(visitor);
                 }
             }
         }