diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 89576b763d..190d85fd51 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -48,9 +48,11 @@ jobs: - name: Test dtype with Python emulator run: DEBUG=2 PYTHON=1 python3 test/test_dtype.py - name: Test ops with Python emulator - run: DEBUG=2 PYTHON=1 python3 -m pytest test/test_ops.py -k "not (test_split or test_simple_cumsum or test_cumsum or test_einsum or test_dot_1d or test_big_gemm or test_broadcastdot or test_multidot or test_var_axis or test_std_axis or test_broadcast_full or test_broadcast_partial or test_simple_conv3d or test_dilated_conv_transpose2d or test_simple_conv_transpose3d or test_large_input_conv2d or test_maxpool2d_simple or test_maxpool2d_bigger_stride or test_avgpool2d or test_cat or test_scaled_product_attention or test_scaled_product_attention_causal)" + run: DEBUG=2 PYTHON=1 python3 -m pytest test/test_ops.py -k "not (test_split or test_simple_cumsum or test_cumsum or test_einsum or test_dot or test_dot_1d or test_big_gemm or test_broadcastdot or test_multidot or test_var_axis or test_std_axis or test_broadcast_full or test_broadcast_partial or test_simple_conv3d or test_dilated_conv_transpose2d or test_simple_conv_transpose3d or test_large_input_conv2d or test_maxpool2d or test_maxpool2d_simple or test_maxpool2d_bigger_stride or test_avgpool2d or test_cat or test_scaled_product_attention or test_scaled_product_attention_causal)" --durations=20 - name: Test symbolic with Python emulator run: PYTHONPATH=. PYTHON=1 python3 test/test_symbolic_ops.py + - name: test_linearizer_failures with Python emulator + run: PYTHONPATH=. PYTHON=1 python3 -m pytest -rA test/test_linearizer_failures.py::TestLinearizerFailures::test_failure_1 linter: name: Linters diff --git a/tinygrad/runtime/ops_python.py b/tinygrad/runtime/ops_python.py index 320f17e319..ea0e4cc530 100644 --- a/tinygrad/runtime/ops_python.py +++ b/tinygrad/runtime/ops_python.py @@ -203,7 +203,7 @@ class PythonProgram: class PythonCompiler(Compiler): linearizer_opts = LinearizerOptions("METAL", has_tensor_cores=True) if getenv("EMULATE_METAL") else \ - (LinearizerOptions("HIP", has_tensor_cores=True) if getenv("EMULATE_HIP") else LinearizerOptions()) + (LinearizerOptions("HIP", has_tensor_cores=True) if getenv("EMULATE_HIP") else LinearizerOptions("PYTHON")) def render(self, name:str, uops:List[UOp]) -> str: lops = [(u.uop, u.dtype, [uops.index(v) for v in u.vin], u.arg) for u in uops] return base64.b64encode(pickle.dumps(lops)).decode()