Skip to content

Commit cb0c494

Browse files
No public description
PiperOrigin-RevId: 966210207
1 parent d40af7b commit cb0c494

5 files changed

Lines changed: 26 additions & 2 deletions

File tree

maldoca/js/ir/BUILD

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
1616
load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
1717
load("@rules_cc//cc:cc_library.bzl", "cc_library")
18+
load("@rules_python//python:defs.bzl", "py_binary")
1819
load("//maldoca/js:allowed_users.bzl", "JSIR_ALLOWED_USERS")
1920

2021
licenses(["notice"])
@@ -378,3 +379,14 @@ filegroup(
378379
"@llvm-project//llvm:FileCheck",
379380
],
380381
)
382+
383+
py_binary(
384+
name = "generate_tests",
385+
srcs = ["generate_tests.py"],
386+
data = [
387+
":jsir_gen",
388+
],
389+
deps = [
390+
"@pypi//tqdm",
391+
],
392+
)

maldoca/js/ir/analyses/constant_propagation/analysis.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -193,10 +193,10 @@ void JsirConstantPropagationAnalysis::VisitOpDefault(
193193

194194
// Merge in the result of the fold, either a constant or a value.
195195
mlir::OpFoldResult fold_result = std::get<1>(it);
196-
if (auto attr = fold_result.dyn_cast<mlir::Attribute>()) {
196+
if (auto attr = llvm::dyn_cast<mlir::Attribute>(fold_result)) {
197197
result.Join(JsirConstantPropagationValue{attr});
198198
} else {
199-
auto result_value = fold_result.get<mlir::Value>();
199+
auto result_value = llvm::cast<mlir::Value>(fold_result);
200200
auto result_state_ref = GetStateAt(result_value);
201201
result.Join(result_state_ref.value());
202202
}

maldoca/js/ir/generate_tests.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,18 @@
55
`test_config.json` found in the scanned directory.
66
77
Usage:
8+
# Option 1: Direct Python
89
python3 maldoca/js/ir/generate_tests.py <directory_to_scan>
910
11+
# Option 2: Blaze build and execute binary
12+
bazel build //maldoca/js/ir:generate_tests
13+
./bazel-bin/maldoca/js/ir/generate_tests <directory_to_scan>
14+
1015
Example:
1116
python3 maldoca/js/ir/generate_tests.py maldoca/js/ir
17+
# or
18+
bazel build //maldoca/js/ir:generate_tests
19+
./bazel-bin/maldoca/js/ir/generate_tests maldoca/js/ir
1220
"""
1321

1422
import argparse

requirements.in

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
# TODO(tzx): Seems unnecessary.
22
# lit==18.1.8
3+
tqdm

requirements.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,6 @@
44
#
55
# pip-compile requirements.in
66
#
7+
tqdm==4.67.1
8+
# via -r requirements.in
9+

0 commit comments

Comments
 (0)