Code performance optimization based on an information-enhanced diffusion model

A computer program product, comprising:

• one or more tangible computer-readable storage media and program instructions stored on at least one of the one or more tangible computer-readable storage media, the program instructions executable by a processor, the program instructions comprising:
• obtaining a given segment of computer code;
• modifying the given segment of computer code to produce one or more inefficient versions of the given segment of computer code in comparison to the given segment of computer code;
• generating a code parse tree for the given segment of computer code and each inefficient version of the given segment of computer code;
• generating model embeddings based on the generated code parse trees;
• and
• training a diffusion model based on the generated model embeddings.

A system comprising:

• a memory;
• and
• at least one processor, coupled to said memory, and operative to perform operations comprising:
• obtaining a given segment of computer code;
• modifying the given segment of computer code to produce one or more inefficient versions of the given segment of computer code in comparison to the given segment of computer code;
• generating a code parse tree for the given segment of computer code and each inefficient version of the given segment of computer code;
• generating model embeddings based on the generated code parse trees;
• and
• training a diffusion model based on the generated model embeddings.