mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
Merge pull request #26369 from justinjfu:sourcemap_win
PiperOrigin-RevId: 724032624
This commit is contained in:
commit
1b7b04f7db
@ -11,6 +11,8 @@
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
import sys
|
||||
|
||||
from absl.testing import absltest
|
||||
from absl.testing import parameterized
|
||||
from jax import numpy as jnp
|
||||
@ -20,6 +22,10 @@ from jax.experimental import source_mapper
|
||||
|
||||
class SourceMapperTest(jtu.JaxTestCase):
|
||||
|
||||
def setUp(self):
|
||||
if sys.platform == "win32":
|
||||
self.skipTest("Only works on non-Windows platforms")
|
||||
|
||||
def test_jaxpr_pass(self):
|
||||
def jax_fn(x, y):
|
||||
return x + y
|
||||
|
Loading…
x
Reference in New Issue
Block a user